剑指offer:复杂链表的复制

题目描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点）。

/*
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 NULL;
        }

        RandomListNode* p=pHead;


        while(p){
            RandomListNode *newp= new RandomListNode(p->label);

            RandomListNode* pNext=p->next;

            p->next=newp;

            newp->next=pNext;

            p=pNext;
        }


        p=pHead;

        while(p){
            RandomListNode* newp=p->next;

            if(p->random){
                newp->random=p->random->next;
            }

            p=newp->next;
        }


        //拆分部分很关键！

        RandomListNode *newHead=pHead->next;
        p=pHead;

        while(p->next){

            RandomListNode *next=p->next;
            p->next=next->next;
            p=next;

        }

        return newHead;
    }
};