【剑指Offer系列35】复杂链表的复制

题目

请实现 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 。

代码

Python

"""
# 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
"""
# 思路：
# 将链表看作图，采用dfs求解，借助哈希表以空间换时间
# 复杂度：
# O(N)
class Solution:
    def copyRandomList(self, head: 'Node') -> 'Node':
        def dfs(head):
            if not head: return None
            if head in visited: return visited[head]
            copy=Node(head.val,None,None) # 创建新结点
            visited[head]=copy # 存入哈希表
            copy.next=dfs(head.next) # 递归拷贝next结点
            copy.random=dfs(head.random) # 递归拷贝random结点
            return copy
        visited={}
        return dfs(head)

C++

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;
    
    Node(int _val) {
        val = _val;
        next = NULL;
        random = NULL;
    }
};
*/
class Solution {
    map<Node*,Node*> visited;
public:
    Node* copyRandomList(Node* head) {
        return dfs(head);
    }
    
    Node* dfs(Node* head) {
        if (head==NULL) return NULL;
        if (visited[head]!=NULL) return visited[head];
        Node* copy=new Node(head->val);
        visited[head]=copy;
        copy->next=dfs(head->next);
        copy->random=dfs(head->random);
        return copy;
    }
};