LeetCode-138. 复制带随机指针的链表 Python3版本

138. 复制带随机指针的链表

---

给定一个链表，每个节点包含一个额外增加的随机指针，该指针可以指向链表中的任何节点或空节点。

要求返回这个链表的深拷贝。 

示例：

输入： {"$id":"1","next":{"$id":"2","next":null,"random":{"$ref":"2"},"val":2},"random":{"$ref":"2"},"val":1}

解释： 节点 1 的值是 1，它的下一个指针和随机指针都指向节点 2 。 节点 2 的值是 2，它的下一个指针指向 null，随机指针指向它自己。

提示：

1. 你必须返回给定头的拷贝作为对克隆列表的引用。

解题思路：先遍历一遍链表，复制生成普通的不带随机指针的新链表，在过程中使用字典记录每个旧节点与新节点的一一对应关系，然后再遍历一次旧链表，根据字典的信息生成新链表的随机指针。

Python3代码如下：

"""
# Definition for a Node.
class Node:
    def __init__(self, val, next, random):
        self.val = val
        self.next = next
        self.random = random
"""
class Solution:
    def copyRandomList(self, head: 'Node') -> 'Node':
        d = {}
        dummy = Node(0,None,None)
        newhead = dummy
        pointer = head
        while pointer:
            newpointer = Node(pointer.val,None,None)
            d[pointer] = newpointer
            newhead.next = newpointer
            newhead = newhead.next
            pointer = pointer.next
        pointer = head
        while pointer:
            if pointer.random:
                d[pointer].random = d[pointer.random]
            pointer = pointer.next
        return dummy.next