题目描述:
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
题解:难点在于如果用dummy head(dummy也等同于pre)的话,pre.random不知道该如何定义。所以我们利用哈希表的查询特点(哈希表搜索快),考虑构建 原链表节点 和 新链表对应节点 的键值对映射关系,再遍历构建新链表各节点的 next 和 random 引用指向即可。
"""
# 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
dic = {
}
cur = head
# 复制各节点,并建立 “原节点 -> 新节点” 的 Map 映射
while cur:
dic[cur] = Node(cur.val)
cur = cur.next
cur = head
# 构建新节点的 next 和 random 指向
while cur:
dic[cur].next = dic.get(cur.next)
dic[cur].random = dic.get(cur.random)
cur = cur.next
# 返回新链表的头节点
return dic[head]