题目:
给你一个长度为 n
的链表,每个节点包含一个额外增加的随机指针 random
,该指针可以指向链表中的任何节点或空节点。
构造这个链表的 深拷贝。 深拷贝应该正好由 n
个 全新 节点组成,其中每个新节点的值都设为其对应的原节点的值。新节点的 next
指针和 random
指针也都应指向复制链表中的新节点,并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点 。
解法一(复制拼接再拆分):
先不考虑random指针的部分,仅将每个节点进行复制紧跟在其后面,再复制random指针的部分。
- 假设原链表如图
- 依次复制每个节点
- 复制random指针
- 将原链表和复制链表拆分出来
代码如下:(参考作者Krahets)
class Solution {
public:
Node* copyRandomList(Node* head) {
if(head == nullptr) return nullptr;
Node* cur = head;
// 1. 复制各节点
while(cur != nullptr) {
Node* tmp = new Node(cur->val);
tmp->next = cur->next;
cur->next = tmp;
cur = tmp->next;
}
// 2. 构建各新节点的 random 指向
cur = head;
while(cur != nullptr) {
if(cur->random != nullptr)
cur->next->random = cur->random->next;
cur = cur->next->next;
}
// 3. 拆分链表
cur = head->next;
Node* pre = head, *res = head->next;
while(cur->next != nullptr) {
pre->next = pre->next->next;
cur->next = cur->next->next;
pre = pre->next;
cur = cur->next;
}
pre->next = nullptr;
return res; // 返回新链表头节点
}
};
解法二(哈希表法):
建立一个哈希表来存储每个节点的next及random的映射关系。
class Solution {
public:
Node* copyRandomList(Node* head) {
unordered_map<Node*,Node*> h;
//将链表的节点值复制进哈希表
for(Node* cur=head;cur;cur=cur->next){
h[cur]=new Node(cur->val);
}
//建立next和random的映射
for(Node* cur=head;cur;cur=cur->next){
if(cur->next) h[cur]->next=h[cur->next];
if(cur->random) h[cur]->random=h[cur->random];
}
return h[head];
}
};