输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点)。
/*
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==NULL)
return pHead;
cloneNext(pHead);
cloneRandom(pHead);
pHead=separate(pHead);
return pHead;
}
private:
void cloneNext(RandomListNode* head){
while(head!=NULL){
RandomListNode* cloneNode=new RandomListNode(head->label);
cloneNode->next=head->next;
head->next=cloneNode;
head=cloneNode->next;
}
}
void cloneRandom(RandomListNode* head){
while(head!=NULL){
if(head->random!=NULL)
head->next->random=head->random->next;
head=head->next->next;
}
}
RandomListNode* separate(RandomListNode* head){
RandomListNode* cloneNode=head->next;
RandomListNode* cloneHead=head->next;
while(cloneNode->next!=NULL){
head->next=cloneNode->next;
cloneNode->next=cloneNode->next->next;
head=head->next;
cloneNode=cloneNode->next;
}
head->next=NULL;
return cloneHead;
}
};