给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的 深拷贝。
我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示:
/*
// 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) {
if (!head) return NULL;
Node *dummy = new Node(-1), *cur = dummy;
Node *p = head;
unordered_map<Node*, Node*> mp;
while (p) {
cur->next = new Node(p->val);
mp[p] = cur->next;
cur = cur->next;
p = p->next;
}
p = head;
cur = dummy->next;
while (p) {
cur->random = mp[p->random];
cur = cur->next;
p = p->next;
}
Node *res = dummy->next;
delete dummy;
return res;
}
};