35.复杂链表的复制
给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的 深拷贝。
我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示:
val:一个表示 Node.val 的整数。
random_index:随机指针指向的节点索引(范围从 0 到 n-1);如果不指向任何节点,则为 null 。
输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]
输入:head = [[1,1],[2,1]]
输出:[[1,1],[2,1]]
输入:head = [[3,null],[3,0],[3,null]]
输出:[[3,null],[3,0],[3,null]]
就地复制
- 复制节点,同时将复制节点链接到原节点后面,如A->B->C 变为 A->A’->B->B’->C->C’。
- 设置节点random值。
- 将复制链表从原链表分离。
class Solution {
public:
Node* copyRandomList(Node* head) {
if (head == nullptr)
return head;
Node* p1 = head;
Node* cur = head;
while(cur)
{
Node* node = new Node(cur->val);
cur = cur->next;
p1->next = node;
node->next = cur;
p1 = cur;
}
cur = head;
while(cur)
{
cur->next->random = cur->random ? cur->random->next : nullptr;
cur = cur->next->next;
}
cur = head;
p1 = head->next;
Node* res = p1;
while(cur)
{
cur->next = cur->next->next;
cur = cur->next;
p1->next = cur ? cur->next : nullptr;
p1 = p1->next;
}
return res;
}
};