Linked List Cycle II @ LeetCode (Python)
Climbing Stairs @ LeetCode (Python)

Copy List with Random Pointer @ LeetCode (Python)

kitt posted @ 2014年3月05日 23:40 in LeetCode , 2120 阅读

来自1楼的光芒, 不解释。Thanks for the link in 1st floor.

# class RandomListNode:
#     def __init__(self, x):
#         self.label = x
#         self.next = None
#         self.random = None

class Solution:
    # @param head, a RandomListNode
    # @return a RandomListNode
    def copyRandomList(self, head):
        if head == None: return None
        # change N1->N2->N3... to N1->newN1->N2->newN2->N3->newN3...
        p, num = head, 0
        while p:
            t = RandomListNode(p.label)
            t.next, p.next, t.random = p.next, t, p.random
            p, num = t.next, num + 1
        # let newNode.random point to correct node
        p = head.next
        for i in xrange(num):
            if p.random: p.random = p.random.next
            if p and p.next: p = p.next.next
        # restore original list and get new list
        p1, p2, newHead = head, head.next, head.next
        for i in xrange(num - 1):
            p1.next, p2.next = p1.next.next, p2.next.next
            p1, p2 = p1.next, p2.next
        p1.next, p2.next = None, None
        return newHead    
Avatar_small
有问题 说:
2014年4月27日 21:51

这道题目有constant extra space的解法,很有意思。可以参考
https://github.com/starforever/leet-code/blob/master/Copy%20List%20with%20Random%20Pointer/Solution.java

Avatar_small
kitt 说:
2014年4月30日 20:04

@有问题: 这个解法真是犀利, 巧妙啊!


登录 *


loading captcha image...
(输入验证码)
or Ctrl+Enter