【LeetCode138】复制带随机指针的链表(dfs+哈希unordered_map)

简介: 0 <= n <= 1000-10000 <= Node.val <= 10000Node.random 为空(null)或指向链表中的节点。

一、题目

image.png

示例1:


输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]

示例2:

image.png

输入:head = [[1,1],[2,1]]
输出:[[1,1],[2,1]]

示例3:

image.png

输入:head = [[3,null],[3,0],[3,null]]
输出:[[3,null],[3,0],[3,null]]

示例4:

输入:head = []
输出:[]
解释:给定的链表为空(空指针),因此返回 null。

提示:

  • 0 <= n <= 1000
  • -10000 <= Node.val <= 10000
  • Node.random 为空(null)或指向链表中的节点。

二、思路

有的同学看到题目的“深拷贝”,大呼很快啊,python中这个深拷贝简单哇,一行代码搞定!二话不说提交:

"""
# 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: 'Optional[Node]') -> 'Optional[Node]':
        return copy.deepcopy(head)

确实可以通过:

image.png

不过面试时面试官肯定不会满意hhh。


从算法角度更好做法是回溯+哈希。使用哈希表unorder_map,存放的key是原链表的节点,对应的value值是key节点的复制节点。利用count查看是否有对应的键key。


利用DFS从head开始遍历整个链表:


(1)每个节点含有随机指针random和常规的next指针,这个随机指针指向的节点是“任意的”,即可能是还未复制出来的节点,那这时候就需要复制操作,这个复制过程即第二点。


(2)利用哈希表的count查看当前node(即key)是否在哈希表中,即查看当前node是否有“复制节点”啦。如果木有则需要new一个节点存放在哈希表中,即存入以node为key的对应的value中,即建立当前结点和“复制结点”的映射。


(3)然后递归调用当前节点node的随机指针random和常规的next指针指向的节点,操作即又进入第二点的过程,进行复制,然后让复制节点的 next 指针和 random 指针指向这两个复制的节点。最后返回已经被访问过的节点的复制节点。

image.png

(4)在回溯搜索过程中,如果当前正在搜索的节点出现在了哈希表中,就说明我们已经遍历完了整个链表,此时就可以直接从哈希表中取出复制后的节点的指针并返回。


image.png

具体过程如下:

1、从 head 节点开始 dfs 遍历整个图。

2、复制当前节点 node,并使用哈希表hash存贮源节点到复制节点之间的映射。

3、递归调用当前节点 node 的后继节点和随机指针指向的节点,并进行复制,然后让复制节点的 next 指针和 random 指针指向这两个复制的节点。

4、最后返回已经被访问过的节点的复制节点。


时间复杂度分析: O(n),其中 n 是链表的长度 。

空间复杂度分析: O(n),其中 n 是链表的长度 。


三、代码

class Solution {
public:
    unordered_map<Node*, Node*> hash;
    Node* copyRandomList(Node* head) 
    {
        if (head == NULL)  return NULL;
        return dfs(head);   
    }
    Node* dfs(Node* node)
    {
        if(node == NULL) return NULL;
        //node节点已经被访问过了,直接从哈希表hash中取出对应的克隆节点返回。
        if(hash.count(node))  return hash[node]; 
        Node* clone = new Node(node->val);    //复制节点
        hash[node] = clone;                   //建立源节点到复制节点的映射   
        clone->next = dfs(node->next);        //复制边  
        clone->random = dfs(node->random);
        return clone;
    }
};


相关文章
|
6月前
|
算法
LeetCode刷题---21.合并两个有序链表(双指针)
LeetCode刷题---21.合并两个有序链表(双指针)
|
6月前
|
算法
LeetCode刷题---19. 删除链表的倒数第 N 个结点(双指针-快慢指针)
LeetCode刷题---19. 删除链表的倒数第 N 个结点(双指针-快慢指针)
|
1月前
链表指针的传参,传值和传地址
本文讨论了链表操作中指针传参的问题,特别是指针的传值与传地址的区别,并提供了修正代码,以确保链表插入操作能正确地修改指针指向的地址。
16 1
链表指针的传参,传值和传地址
|
5月前
|
存储 算法 数据挖掘
python 数学+减治、下一个排列法、DFS回溯法实现:第 k 个排列【LeetCode 题目 60】
python 数学+减治、下一个排列法、DFS回溯法实现:第 k 个排列【LeetCode 题目 60】
|
6月前
|
存储 C语言
用指针处理链表
用指针处理链表
59 3
|
1月前
|
C语言
无头链表二级指针方式实现(C语言描述)
本文介绍了如何在C语言中使用二级指针实现无头链表,并提供了创建节点、插入、删除、查找、销毁链表等操作的函数实现,以及一个示例程序来演示这些操作。
22 0
|
4月前
【数据结构OJ题】复制带随机指针的链表
力扣题目——复制带随机指针的链表
51 1
【数据结构OJ题】复制带随机指针的链表
|
5月前
|
算法
【经典LeetCode算法题目专栏分类】【第9期】深度优先搜索DFS与并查集:括号生成、岛屿问题、扫雷游戏
【经典LeetCode算法题目专栏分类】【第9期】深度优先搜索DFS与并查集:括号生成、岛屿问题、扫雷游戏
|
5月前
|
存储 SQL 算法
LeetCode题目100:递归、迭代、dfs使用栈多种算法图解相同的树
LeetCode题目100:递归、迭代、dfs使用栈多种算法图解相同的树
|
6月前
[leetcode~dfs]1261. 在受污染的二叉树中查找元素
[leetcode~dfs]1261. 在受污染的二叉树中查找元素
[leetcode~dfs]1261. 在受污染的二叉树中查找元素