当前位置:网站首页>[牛客网刷题 Day4] JZ35 复杂链表的复制

[牛客网刷题 Day4] JZ35 复杂链表的复制

2022-07-05 08:39:00 strawberry47

题目

在这里插入图片描述

解答:

看不懂题目,好像输入和输出一样??
哦!原来是每个节点后面跟了一个next指针和random指针哦

想法:
用一个list存储所有的random节点,再把他们加到常规链表后面。
但是null节点没法加next了,而且一开始dummy=pHead,那就一直有random存在。。。

啊啊w(゚Д゚)w,原来我搞错题意了!!题目是要求深拷贝,并不是把节点串起来呀!

思路:建立一个字典,key是当前node的值,value是random的值;然后遍历这个字典。

class RandomListNode:
    def __init__(self, x):
        self.label = x
        self.next = None
        self.random = None

class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        if pHead is None:
            return None
        res = RandomListNode(0)
        node_dict = dict()

        cur = pHead
        while cur:
            node_dict[cur.label] = cur.random
            cur = cur.next

        dummy = res
        for node,random1 in node_dict.items():
            res.next = RandomListNode(node)
            res = res.next
            res.random = random1

        return dummy.next

参考答案:

思路有点类似,也是创建了一个哈希表,key是当前节点,value是节点的值;
然后再遍历一遍原始链表,把原来的next和random都取出来

class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        if  pHead is None:
            return None   
        dict = {
    }
        cur = pHead
        while cur:
            dict[cur] = RandomListNode(cur.label)
            cur = cur.next
        cur = pHead
        while cur:
            dict[cur].next = dict.get(cur.next)
            dict[cur].random = dict.get(cur.random)
            cur = cur.next
        return dict[pHead]
原网站

版权声明
本文为[strawberry47]所创,转载请带上原文链接,感谢
https://blog.csdn.net/strawberry47/article/details/125581092