题目26:复杂链表的复制(leetcode链接:https://leetcode-cn.com/problems/fu-za-lian-biao-de-fu-zhi-lcof/)
问题描述
给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。要求返回这个链表的 深拷贝。 我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示:val:一个表示 Node.val 的整数。random_index:随机指针指向的节点索引(范围从 0 到 n-1);如果不指向任何节点,则为 null 。
示例 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]]
示例 4:输入:head = [] 输出:[]
解释:给定的链表为空(空指针),因此返回 null。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/copy-list-with-random-pointer
解题思路
代码描述
class Solution {
public:
Node* copyRandomList(Node* head) {
if(head == NULL)
return NULL;
//为每个节点创建一个新节点,链接在原来节点的后边,并且将新节点的random置为NULL
Node* curr = head;
while(curr)
{
Node* newNode = (Node*)malloc(sizeof(Node));
newNode->val = curr->val;
newNode->next = curr->next;
curr->next = newNode;
newNode->random = NULL;
curr = newNode->next;
}
//新节点的random是原来节点的random的next节点
curr = head;
while(curr && curr->next)
{
if(curr->random == NULL)
{
curr->next->random = NULL;
}
else
{
curr->next->random = curr->random->next;
}
curr = curr->next->next;
}
//分离新节点和旧节点
curr= head;
Node* newHead = (Node*)malloc(sizeof(Node));
Node* tail = newHead;
while(curr && curr->next)
{
tail->next = curr->next;
curr->next = curr->next->next;
tail = tail->next;
curr = curr->next;
}
curr = newHead->next;
free(newHead);
return curr;
}
};