[牛客网刷题 Day4] JZ35 复杂链表的复制

题目

解答：

看不懂题目，好像输入和输出一样？？
哦！原来是每个节点后面跟了一个next指针和random指针哦

想法：
用一个list存储所有的random节点，再把他们加到常规链表后面。
但是null节点没法加next了，而且一开始dummy=pHead，那就一直有random存在。。。

啊啊w(ﾟДﾟ)w，原来我搞错题意了！！题目是要求深拷贝，并不是把节点串起来呀！

思路：建立一个字典，key是当前node的值，value是random的值；然后遍历这个字典。

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

class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        if pHead is None:
            return None
        res = RandomListNode(0)
        node_dict = dict()

        cur = pHead
        while cur:
            node_dict[cur.label] = cur.random
            cur = cur.next

        dummy = res
        for node,random1 in node_dict.items():
            res.next = RandomListNode(node)
            res = res.next
            res.random = random1

        return dummy.next

参考答案：

思路有点类似，也是创建了一个哈希表，key是当前节点，value是节点的值；
然后再遍历一遍原始链表，把原来的next和random都取出来

class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        if  pHead is None:
            return None   
        dict = {}
        cur = pHead
        while cur:
            dict[cur] = RandomListNode(cur.label)
            cur = cur.next
        cur = pHead
        while cur:
            dict[cur].next = dict.get(cur.next)
            dict[cur].random = dict.get(cur.random)
            cur = cur.next
        return dict[pHead]