python算法日记(链表系列)_leetcode 138. 复制带随机指针的链表

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

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

我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示:

val:一个表示 Node.val 的整数。
random_index:随机指针指向的节点索引(范围从 0 到 n-1);如果不指向任何节点,则为  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。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/copy-list-with-random-pointer

"""
# 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
        deep_dict = {}  #用字典存新旧节点
        old = head
        new = Node(0)
        while(head):  
            key = head
            new.next = Node(head.val) #新节点
            value = new.next
            new = new.next
            deep_dict.update({key:value}) #新旧节点存进字典
            head = head.next
        for key,value in deep_dict.items():
            if key.random:
            # 例value: a' ,key: a ,a.random = c,deep_dict[a.random]=c',所以a'.random = c'
                value.random = deep_dict[key.random]  
        return deep_dict[old]
发布了44 篇原创文章 · 获赞 0 · 访问量 1901

猜你喜欢

转载自blog.csdn.net/weixin_39331401/article/details/104643955