题目描述
来源:牛客网《剑指Offer》
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
解题思路
借鉴他人的图解:
(1)原始复杂链表如图所示,蓝色线条为特殊指针random;
(2)首先复制所有链表结点,并将复制后的结点插入原始链表;
(3)复制原始链表结点的特殊指针random;
(4)将原始链表和复制的链表进行拆分;
/*
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
*/
class Solution {
public:
RandomListNode* Clone(RandomListNode* Head)
{
RandomListNode* p = Head;
//复制所有链表结点,并将复制后的结点插入原始链表;
while(p)
{
//创建新节点
RandomListNode* c = new RandomListNode(p->label);
//插入新节点
RandomListNode* temp = p->next;
p->next = c;
c->next = temp;
p = temp;
}
//复制原始链表结点的特殊指针random;
p = Head;
while(p)
{
//p指向原始结点,p->next是复制结点
if(p->random)
p->next->random = p->random->next;
p = p->next->next;
}
//将原始链表和复制的链表进行拆分;
//虚拟头结点
RandomListNode* dummpy = new RandomListNode(-1);
RandomListNode* cur = dummpy;
for(p = Head; p; p = p -> next)
{
cur -> next = p -> next;//A'
cur = cur -> next;
p -> next = p -> next -> next;
}
return dummpy->next;
}
};