随机链表的复制
题目连接:https://leetcode.cn/problems/copy-list-with-random-pointer/?envType=study-plan-v2&envId=top-100-liked
给你一个长度为 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]]
提示:
0 <= n <= 1000
-10^4 <= Node.val <= 10^4
Node.random
为null
或指向链表中的节点。
思路
这道题理解题意比较困难,这里说明一下题目想要表达的意思。
给你一个链表,链表中的每个节点除了自身 val 值和指向下一个节点的 next 指针,还有一个指向链表中某个节点的随机指针 random。题目要我们做的操作是将这个链表完整的复制下来,注意,不是将原来的链表原原本本地复制一遍,是新创建一个链表,将这个链表的结构仿照原有的列表创建出来。简单来说就是新旧链表对应的每个节点的地址不同。
那么我们应该如何操作才能完成随机链表的复制呢?我们可以借助将链表转换成线性表的思路,实现对链表的复制。
先将原来的链表转换成线性表,然后我们可以通过哈希表建立映射关系:节点在线性表中的下标 -> 节点地址。这样映射之后,我们可以通过线性表的下标访问到节点的 random 所指向的节点。
然后我们再建立另一个映射关系:当前节点下标 -> random节点下标。通过这个映射关系,我们可以将旧链表中的 random 关系复制到新链表中。
还要注意有一个细节需要处理:旧链表中节点的 random 可能指向 nullptr,所以需要特判一下,避免和下标为 0 的节点重复导致过不了样例。代码如下
/*
// Definition for a Node.
class Node {
public:
int val;
Node* next;
Node* random;
Node(int _val) {
val = _val;
next = NULL;
random = NULL;
}
};
*/
class Solution {
public:
Node* copyRandomList(Node* head) {
int len = 0; // 链表长度
vector<Node*> List1; // 转换成线性表
// 将当前节点的地址和下标对应起来
unordered_map<Node*, int> map1;
Node *p = head;
// 创建线性表
while (p != nullptr) {
map1.insert({p, len});
len++;
List1.push_back(p);
p = p->next;
}
Node *dummy = new Node(0);
p = dummy;
Node *q; // 用于创建新节点
// 第一个是当前下标,第二个是random的下标
unordered_map<int, int> map2;
vector<Node*> List2; // 存放新建的链表
// 创建新链表
for (int i = 0; i < len; ++i) {
// 特判random指向空
if (List1[i]->random == nullptr) {
map2.insert({i, -1});
}
map2.insert({i, map1[List1[i]->random]});
// 创建新节点
q = new Node(List1[i]->val);
List2.push_back(q);
// 连接到新链表尾部
p->next = q;
p = p->next;
}
// 更新新链表中的random
for (int i = 0; i < len; ++i) {
if (map2[i] == -1) List2[i]->random = nullptr;
else List2[i]->random = List2[map2[i]];
}
return dummy->next;
}
};