Leetcode 138.复制带随机指针的链表(medium)

题目

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

构造这个链表的 深拷贝。 深拷贝应该正好由 n 个 全新 节点组成,其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点,并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点 。

例如,如果原链表中有 X 和 Y 两个节点,其中 X.random --> Y 。那么在复制链表中对应的两个节点 x 和 y ,同样有 x.random --> y 。

返回复制链表的头节点。

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

val:一个表示 Node.val 的整数。
random_index:随机指针指向的节点索引(范围从 0 到 n-1);如果不指向任何节点,则为  null 。
你的代码 只 接受原链表的头节点 head 作为传入参数。

示例 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

1. 原始解法

分为两步,第一步复制链表中除了random指针之外的内容;第二步复制random指针。第二步中,针对原链表中每个节点,使用循环以找到题目中所表示的random_index,再在新链表中使用循环建立正确的random指针。

结果:通过测试用例,用时68ms,内存消耗15.6MB

代码:

class Solution:
    def copyRandomList(self, head: 'Node') -> 'Node':
        if not head:
            return None
        new_head = Node(head.val)
        p = head
        q = new_head
        while p.next:
            p = p.next
            q.next = Node(p.val)               
            q = q.next

        
        p = head
        q = new_head        
        while p:
            if p.random:
                count = 0
                r = head
                while r != p.random:
                    r = r.next
                    count += 1

                r = new_head
                for i in range(count):
                    r = r.next
                q.random = r

                        
            p = p.next
            q = q.next

        return new_head

2. 改进

通过学习题解,领悟到可以将原列表中的节点与复制的新节点在一个字典中建立一一对应的关系。假设原链表的一个节点为p,对应节点为q,那么q.random实际上就应该是字典中p.random所对应的值。

结果:通过测试用例,用时32ms,内存使用15.7MB

代码:

def copyRandomList(self, head: 'Node') -> 'Node':
        if not head:
            return None     
        nodes = {}

        new_head = Node(head.val)
        p = head
        q = new_head
        while p.next:
            nodes[p] = q  
            p = p.next                      
            q.next = Node(p.val) 
            q = q.next
        
        nodes[p] = q
        
        
        p = head
        q = new_head      
        while p:            
            if p.random:
                q.random = nodes[p.random]            
                        
            p = p.next
            q = q.next

        return new_head

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值