题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
/*
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
*/
class Solution {
public:
//非常巧妙的方法,在原来链表的中间插入新的节点,使新旧链表交错连接着,这样就能很轻易的通过
//旧链表的随机指针的相对位置来连接新链表了
RandomListNode* Clone(RandomListNode* pHead)
{
if (pHead == nullptr)
return nullptr;
RandomListNode* p = pHead;
//在旧链表之间插入新链表
while (p != nullptr)
{
RandomListNode* tp = new RandomListNode(p->label);
tp->next = p->next;
p->next = tp;
p = tp->next;
}
p = pHead;
RandomListNode* p2, *pHead2;
p2 = pHead2 = p->next;
//根据旧链表的相对位置连接好新链表的随机指针
while (true)
{
if (p->random == nullptr)
p2->random = nullptr;
else
p2->random = p->random->next;
if (p2->next == nullptr)
break;
p = p2->next;
p2 = p->next;
}
p = pHead, p2 = pHead2;
//把新旧链表拆开
while (true)
{
p->next = p2->next;
if (p2->next == nullptr)
break;
p2->next = p2->next->next;
p = p->next;
p2 = p2->next;
}
return pHead2;
}
};