LeetCode-138. 复制带随机指针的链表 Python3版本

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


给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。

要求返回这个链表的深拷贝。 

示例:

LeetCode-138. 复制带随机指针的链表 Python3版本_第1张图片

LeetCode-138. 复制带随机指针的链表 Python3版本_第2张图片输入: {"$id":"1","next":{"$id":"2","next":null,"random":{"$ref":"2"},"val":2},"random":{"$ref":"2"},"val":1}

解释: 节点 1 的值是 1,它的下一个指针和随机指针都指向节点 2 。 节点 2 的值是 2,它的下一个指针指向 null,随机指针指向它自己。

提示:

  1. 你必须返回给定头的拷贝作为对克隆列表的引用。

解题思路:先遍历一遍链表,复制生成普通的不带随机指针的新链表,在过程中使用字典记录每个旧节点与新节点的一一对应关系,然后再遍历一次旧链表,根据字典的信息生成新链表的随机指针。

Python3代码如下:

"""
# Definition for a Node.
class Node:
    def __init__(self, val, next, random):
        self.val = val
        self.next = next
        self.random = random
"""
class Solution:
    def copyRandomList(self, head: 'Node') -> 'Node':
        d = {}
        dummy = Node(0,None,None)
        newhead = dummy
        pointer = head
        while pointer:
            newpointer = Node(pointer.val,None,None)
            d[pointer] = newpointer
            newhead.next = newpointer
            newhead = newhead.next
            pointer = pointer.next
        pointer = head
        while pointer:
            if pointer.random:
                d[pointer].random = d[pointer.random]
            pointer = pointer.next
        return dummy.next

LeetCode-138. 复制带随机指针的链表 Python3版本_第3张图片

你可能感兴趣的:(LeetCode)