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

题目

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

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

我们用一个由 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。

提示：

-10000 <= Node.val <= 10000
Node.random 为空（null）或指向链表中的节点。
节点数目不超过 1000 。

解题思路

额外开辟o(n)空间版：
最简单的思路，对每个新老节点保存一个映射，然后根据老节点的random找到新节点的映射，给新节点的random赋值

额外空间o(1)版：
其实上面的思路，是需要额外空间保存一个新老节点的映射。如果直接用原节点的next来代替新老节点的映射则可以省去额外空间。

考虑把复制好的新节点放在老节点后面，即：
原先链表：1 -> 2 -> 3，新链表：1 -> 1' -> 2 -> 2' -> 3 -> 3'
保存好映射后，再考虑random指针，观察可以发现，通过对老节点p进行p.next.random = p.random.next操作即可
最后再从链表中剥离出来新链表

代码

额外空间版：

"""
# 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 copyRandomList(self, head: 'Node') -> 'Node':
        old2new = {None: None}
        p = head
        ret_head = Node(-1)
        new_p = ret_head
        while p:
            new_p.next = Node(p.val)
            new_p = new_p.next
            old2new[p] = new_p
            p = p.next
        # random pointer
        p = head
        new_p = ret_head.next
        while p:
            new_p.random = old2new[p.random]
            p = p.next
            new_p = new_p.next
        return ret_head.next

o(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 copyRandomList(self, head: 'Node') -> 'Node':
        # insert new nodes
        p = head
        while p:
            new_node = Node(p.val)
            new_node.next = p.next
            p.next = new_node
            p = new_node.next
        # p.next.random = p.random.next
        p = head
        while p:
            p.next.random = p.random.next if p.random else None
            p = p.next.next
        # get new nodes
        ret_head = Node(-1)
        prev, p = ret_head, head
        while p:
            prev.next = p.next
            prev = p.next
            p = p.next.next
        return ret_head.next