【链表和数组】剑指 Offer 35. 复杂链表的复制

题目描述
请实现 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]        


猜你喜欢

转载自blog.csdn.net/Rolandxxx/article/details/128889392