leetcode 138. copy-list-with-random-pointer 复制带随机指针的链表 python3

时间：2020-7-9

题目地址：https://leetcode-cn.com/problems/copy-list-with-random-pointer/

题目难度：Medium

题目描述：

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

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

我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示：

val：一个表示 Node.val 的整数。
random_index：随机指针指向的节点索引（范围从 0 到 n-1）；如果不指向任何节点，则为  null 。

 

示例 1：

输入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出：[[7,null],[13,0],[11,4],[10,2],[1,0]]

示例 2：

输入：head = [[1,1],[2,1]]
输出：[[1,1],[2,1]]

示例 3：

输入：head = [[3,null],[3,0],[3,null]]
输出：[[3,null],[3,0],[3,null]]

示例 4：

输入：head = []
输出：[]
解释：给定的链表为空（空指针），因此返回 null。

---

思路1：空间迭代

1、从 head 节点开始遍历链表。首先创造新的 head 拷贝节点。拷贝的节点，我们将该新建节点的引用也放入已访问字典中。

2、random指针：

如果当前节点 ii 的 random 指针指向一个节点 jj 且节点 jj 已经被拷贝过，我们将直接使用已访问字典中该节点的引用而不会新建节点。
如果当前节点 ii 的 random 指针指向
的节点 jj 还没有被拷贝过，我们就对 jj 节点创建对应的新节点，并把它放入已访问节点字典中。

3、next指针：

如果当前节点 ii 的 next 指针指向的节点 jj 在已访问字典中已有拷贝，我们直接使用它的拷贝节点。
如果当前节点 ii 的next 指针指向的节点 jj 还没有被访问过，我们创建一个对应节点的拷贝，并放入已访问字典

代码段1：通过

"""
# Definition for a Node.
class Node:
    def __init__(self, x: int, next: 'Node' = None, random: 'Node' = None):
        self.val = int(x)
        self.next = next
        self.random = random
"""

class Solution:
    def __init__(self):
            self.visited = {}

    def getCloneNode(self, node):
        if node:
            if node in self.visited:
                return self.visited[node]
            else:
                self.visited[node] = Node(node.val, None, None)
                return self.visited[node]
        return None

    def copyRandomList(self, head: 'Node') -> 'Node':
        if not head:
            return 
        
        old_node = head

        new_node = Node(old_node.val, None, None)
        self.visited[old_node] = new_node

        while old_node != None:
            new_node.random = self.getCloneNode(old_node.random)
            new_node.next = self.getCloneNode(old_node.next)

            old_node = old_node.next
            new_node = new_node.next

        return self.visited[head]

总结：

1. 完了完了，迭代也写不出来

---

后续优化：回溯 or 模仿map https://leetcode-cn.com/problems/copy-list-with-random-pointer/solution/fu-zhi-dai-sui-ji-zhi-zhen-de-lian-biao-by-leetcod/