138. 复制带随机指针的链表（左神）

最新推荐文章于 2022-11-10 00:52:40 发布

Grayson Zhang

最新推荐文章于 2022-11-10 00:52:40 发布

阅读量83

点赞数

分类专栏： Leetcode

本文链接：https://blog.csdn.net/qq_51174011/article/details/119077629

版权

Leetcode 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

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

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

构造这个链表的深拷贝。深拷贝应该正好由 n 个全新节点组成，其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点，并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点。

例如，如果原链表中有 X 和 Y 两个节点，其中 X.random --> Y 。那么在复制链表中对应的两个节点 x 和 y ，同样有 x.random --> y 。

返回复制链表的头节点。

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

val：一个表示 Node.val 的整数。
random_index：随机指针指向的节点索引（范围从 0 到 n-1）；如果不指向任何节点，则为 null 。
你的代码只接受原链表的头节点 head 作为传入参数。

示例 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。

提示：
0 <= n <= 1000
-10000 <= Node.val <= 10000
Node.random 为空（null）或指向链表中的节点。

解题：

链表的定义

# 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':
        if not head:
            return None

        node_map = defaultdict()

        p = head
        # 先把节点都创建好，存到map中
        while p:
            new_node = Node(p.val)
            node_map[p] = new_node
            p = p.next
        
        p = head
        # 构建next和random指针
        while p:
            if p.next:
                node_map[p].next = node_map[p.next]

            if p.random:
                next_node = node_map[p.random]
                curr = node_map[p]
                curr.random = next_node
            p = p.next
        return node_map[head]

法二：插入复制节点

在链表中插入要复制的结点，然后逐个更新，最后在剥离出来

class Solution:
    def copyRandomList(self, head: 'Node') -> 'Node':
        if not head:
            return head

        # 复制所有节点，插入原节点的后面
        cur = head
        while cur:
            cur.next = Node(cur.val, cur.next, None)
            cur = cur.next.next

        # 连接所有复制的节点的random指针
        cur = head
        copyHead = head.next
        while cur:
            if cur.random:
                cur.next.random = cur.random.next
            cur = cur.next.next
        
        # 断开原链表与复制链表之间的连接
        cur = head
        cur_ = copyHead
        while cur and cur_:
            cur.next = cur_.next
            cur = cur.next
            if cur:
                cur_.next = cur.next
            cur_ = cur_.next
        return copyHead