138. 复制带随机指针的链表
给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的 深拷贝。
我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示:
- val:一个表示 Node.val 的整数。
- random_index:随机指针指向的节点索引(范围从 0 到 n-1);如果不指向任何节点,则为 null 。
解题思路: 这道题是《剑指offer》上的原题。解这题的关键点是如何计算random的节点。书上的解法和此题的解法均采用维护一个源节点到目标节点的映射,在遍历源链表时,通过它的random节点即可得到目标节点的random节点。
/*
// 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;
}
};