LeetCode 第 138 题:复制带随机指针的链表(Python 代码)

简介: 给定一个长度为 n 的链表,每个节点比普通的节点多了一个额外的随机指针 ramdom,该指针可以指向链表中的任何节点或空节点。构造这个链表的深拷贝。所谓的深拷贝,就是完全生成一个新的对象,内存地址都是不同的,这样改变拷贝之前变量,就不会影响到拷贝的变量。

题目 138. 复制带随机指针的链表 的描述如下,给定一个长度为 n 的链表,每个节点比普通的节点多了一个额外的随机指针 ramdom,该指针可以指向链表中的任何节点或空节点。构造这个链表的深拷贝。所谓的深拷贝,就是完全生成一个新的对象,内存地址都是不同的,这样改变拷贝之前变量,就不会影响到拷贝的变量。

哈希映射 1

首先贴一个比较巧妙的解法,来自用户 skyliuhc,在官方题解下面的评论可以看到,他写的是 Java 的版本,这里我写成 Python 的版本:

主要的思路就是,先用一个循环把新旧链表对应的两个结点捆绑在一个二元组里,然后再用一个循环完成对新链表每个结点的 nextrandom 的赋值

class Solution:
    def copyRandomList(self, head: 'Optional[Node]') -> 'Optional[Node]':
        if not head: return None
        cur = head
        hashmap = {}  # 哈希表
        # 第一个循环,建立映射
        while cur:
            hashmap[cur] = Node(cur.val)  # 新旧链表对应结点映射
            cur = cur.next  # 当前结点往后移动
        # 第二个循环,next 和 random 赋值
        cur = head
        while cur:
            # 因为 Python 的字典中不能用 None 作为键值,所以要做一个特殊判断
            hashmap[cur].next = hashmap[cur.next] if cur.next else None
            hashmap[cur].random = hashmap[cur.random] if cur.random else None
            cur = cur.next
        return hashmap[head]

哈希映射 2

这个是我自己当时做题的时候想的方法,与前面一个的想法类似,也是打算用哈希表来记录新旧链表的结点,但是想法更为复杂了,具体的思路为:

  • 1、先把最普通的链表构建起来,同时记录两个 Hashmap

    • 1.1、Node2Pos:旧链表的每个 Node 的位置
    • 1.2、Pos2Node:新链表每个位置对应的 Node
  • 2、同步遍历新旧两个链表,得到当前位置的旧节点的 random node 对应的位置 pos
  • 3、通过位置 pos 得到新链表的 Node,就知道当前结点的 random 要指向哪里;
class Solution:
    def copyRandomList(self, head: 'Optional[Node]') -> 'Optional[Node]':
        if not head: return None
        Node2Pos = {}  # 旧链表的每个 `Node` 的位置
        Pos2Node = {}  # 新链表每个位置对应的 `Node`
        # 构建新链表
        newPreHead = Node(0)  # 前置结点
        new_ptr = newPreHead  # 新链表的头结点
        old_ptr = head  # 旧链表的头结点 
        idx = 0 # 当前的位置
        while old_ptr: 
            Node2Pos[old_ptr] = idx  # 记录旧结点位置
            new_ptr.next = Node(old_ptr.val)
            new_ptr = new_ptr.next
            Pos2Node[idx] = new_ptr  # 记录当前位置的新结点
            idx += 1  # 位置也要同步往后
            old_ptr = old_ptr.next
        Pos2Node[idx] = None  # 最后一个位置是空指针 None
        
        # 处理 random node
        new_ptr = newPreHead.next
        old_ptr = head
        while old_ptr:
            random_node = old_ptr.random  # 获得旧结点的随机结点
            # 获得旧结点随机结点的位置,如果为随机结点为空代表是最后一个位置,即 idx
            pos = Node2Pos[random_node] if random_node else idx
            node = Pos2Node[pos]  # 获得对应 pos 的结点
            new_ptr.random = node
            new_ptr = new_ptr.next
            old_ptr = old_ptr.next
        
        return newPreHead.next  # 返回结果

回溯 + 哈希表

官方题解:利用回溯法,让每个节点的拷贝操作相互独立,对于当前结点,首先进行拷贝,然后进行「当前节点的后继节点」和「当前节点的随机指针指向的节点」拷贝,拷贝完成后将创建的新节点的指针返回,即可完成当前节点的两指针的赋值。

具体的做法是:

  • 先创建一个哈希表 cachedNode,从头结点开始遍历;
  • 如果结点为空,返回 None
  • 创建一个新的结点,其值与当前遍历的结点值 val 一样;
  • 然后遍历地去对 nextrandom 赋值;
class Solution:
    def __init__(self,):
        self.cachedNode = {}
    
    def copyRandomList(self, head: 'Optional[Node]') -> 'Optional[Node]':
        if not head: return None
        if head not in self.cachedNode.keys():
            headNew = Node(head.val)  # 拷贝新节点
            self.cachedNode[head] = headNew  # 记录到哈希表中
            headNew.next = self.copyRandomList(head.next)
            headNew.random = self.copyRandomList(head.random)
        return self.cachedNode[head]

迭代 + 节点拆分

  • 对于链表 A => B => C,可以将其变成 A => A' => B => B' => C => C',其中 A'A 的拷贝结点
class Solution:
    def copyRandomList(self, head: 'Optional[Node]') -> 'Optional[Node]':
        if not head: return None
        # 将 A => B 变成 A => A' => B => B'
        node = head
        while node:
            nodeNew = Node(node.val)
            nodeNew.next = node.next
            node.next = nodeNew
            node = node.next.next
        
        # 处理 random
        node = head
        while node:
            nodeNew = node.next
            nodeNew.random = node.random.next if node.random else None
            node = node.next.next
            
        # 将 A => A' => B => B' 变成 A => B
        headNew = head.next
        node = head
        while node:
            nodeNew = node.next
            node.next = node.next.next
            nodeNew.next = nodeNew.next.next if nodeNew.next else None
            node = node.next
        
        return headNew
目录
相关文章
|
5月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
130 2
|
5月前
|
算法 Python
【Leetcode刷题Python】 LeetCode 2038. 如果相邻两个颜色均相同则删除当前颜色
本文介绍了LeetCode 2038题的解法,题目要求在一个由'A'和'B'组成的字符串中,按照特定规则轮流删除颜色片段,判断Alice是否能够获胜,并提供了Python的实现代码。
60 3
|
5月前
|
算法 Python
【Leetcode刷题Python】剑指 Offer 33. 二叉搜索树的后序遍历序列
本文提供了一种Python算法,用以判断给定整数数组是否为某二叉搜索树的后序遍历结果,通过识别根节点并递归验证左右子树的值是否满足二叉搜索树的性质。
28 3
|
5月前
|
Python
【Leetcode刷题Python】50. Pow(x, n)
本文介绍了LeetCode第50题"Pow(x, n)"的解法,题目要求实现计算x的n次幂的函数,文章提供了递归分治法的详细解析和Python实现代码。
34 1
|
5月前
|
Python
【Leetcode刷题Python】LeetCode 478. 在圆内随机生成点
本文介绍了LeetCode 478题的解法,题目要求在给定圆的半径和圆心位置的情况下实现在圆内均匀随机生成点的功能,并提供了Python的实现代码。
39 1
|
5月前
|
算法 Python
【Leetcode刷题Python】295. 数据流的中位数
本文介绍了一种使用Python实现的数据结构,用以支持数据流中添加整数并返回当前所有元素的中位数,通过排序列表来计算中位数。
35 1
|
5月前
|
算法 Python
【Leetcode刷题Python】73. 矩阵置零
本文介绍了LeetCode第73题的解法,题目要求在给定矩阵中将所有值为0的元素所在的行和列全部置为0,并提供了一种原地算法的Python实现。
39 0
【Leetcode刷题Python】73. 矩阵置零
|
5月前
|
Python
【Leetcode刷题Python】1467. 两个盒子中球的颜色数相同的概率
本文介绍了LeetCode第50题"Pow(x, n)"的解法,题目要求实现计算x的n次幂的函数,文章提供了递归分治法的详细解析和Python实现代码。
57 0
|
2月前
|
存储 C语言
C语言如何使用结构体和指针来操作动态分配的内存
在C语言中,通过定义结构体并使用指向该结构体的指针,可以对动态分配的内存进行操作。首先利用 `malloc` 或 `calloc` 分配内存,然后通过指针访问和修改结构体成员,最后用 `free` 释放内存,实现资源的有效管理。
161 13
|
3月前
|
C语言
无头链表二级指针方式实现(C语言描述)
本文介绍了如何在C语言中使用二级指针实现无头链表,并提供了创建节点、插入、删除、查找、销毁链表等操作的函数实现,以及一个示例程序来演示这些操作。
41 0