剑指Offer 35：复杂链表的复制

剑指Offer 35：复杂链表的复制

2022年2月20日

---

一、问题描述

请实现 copyRandomList 函数，复制一个复杂链表。在复杂链表中，每个节点除了有一个 next 指针指向下一个节点，还有一个 random 指针指向链表中的任意节点或者 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 。

二、问题分析

• 问题的核心在于如何复制Random指针。
• 考虑构建 原节点 1 -> 新节点 1 -> 原节点 2 -> 新节点 2 -> …… 的拼接链表，如此便可在访问原节点的 random 指向节点的同时找到新对应新节点的 random 指向节点。
• 代码的实现机理如下：（a）将原链表与新链表进行拼接，形成原节点 1 -> 新节点 1 -> 原节点 2 -> 新节点 2 -> ……的形式；（b）将新节点的Random指针指向对应的节点；（c）当遍历链表后，断开原链表和新链表，返回新链表。

（下述示意图参考Leetcode的Krahets大佬）

三、解题代码

"""
# Definition for a Node.
class Node:
    def __init__(self, x, next=None, random=None):
        self.val = int(x)
        self.next = next
        self.random = random
"""
class Solution(object):
    def copyRandomList(self, head):
        """
        :type head: Node
        :rtype: Node
        """
        if not head:
            return
        
        # Merge linked-lists
        curr_node = head
        while curr_node:
            tmp = Node(curr_node.val)
            tmp.next = curr_node.next
            curr_node.next = tmp
            curr_node = tmp.next
        
        # Copy random link
        curr_node = head
        while curr_node:
            if curr_node.random:
                curr_node.next.random = curr_node.random.next
            curr_node = curr_node.next.next
        
        # Seperate to two linked-lists
        prev_node = head
        curr_node = res_node = head.next
        while curr_node.next:
            prev_node.next = prev_node.next.next
            curr_node.next = curr_node.next.next
            prev_node = prev_node.next
            curr_node = curr_node.next
        prev_node.next = None

        return res_node

总结

为实现复制带Random指针的链表，这里将原链表和新链表进行拼接后，对Random指针进行指向，再将链表断开，返回新链表。