複雜鏈表的複製

題目描述

輸入一個複雜鏈表（每個節點中有節點值，以及兩個指針，一個指向下一個節點，另一個特殊指針指向任意一個節點），返回結果爲複製後複雜鏈表的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)
    {
            //第一步，根據原始鏈表的每個節點N創立節點N'
            RandomListNode* pNode=pHead;
            while(pNode!=NULL)
                {
                 RandomListNode* pCloned=new RandomListNode(pHead->label);
                 pCloned->label=pNode->label;
                 pCloned->next=pNode->next;
                 pCloned->random=NULL;
                 pNode->next=pCloned;
                 pNode=pCloned->next;
            }
           // 第二步，複製節點的多餘的指針
            pNode=pHead;
            while(pNode!=NULL)
                {
                 RandomListNode* pCloned=new RandomListNode(pHead->label);
                 pCloned=pNode->next;
                 if(pNode->random!=NULL)
                {
                     //假設pNode指向s，則pCloned指向s的下一個節點s'
                     pCloned->random=pNode->random->next;
                 }
                 pNode=pCloned->next;
            }
            //
            
             //第三步，把長鏈表拆分成兩個鏈表
             pNode=pHead;
             RandomListNode* pClonedHead=NULL;
             RandomListNode* pClonedNode=NULL;
            if(pNode!=NULL)
                {
                 pClonedHead=pClonedNode=pNode->next;
                 pNode->next=pClonedNode->next;
                 pNode=pNode->next;
            }
            while(pNode!=NULL)
                {
                pClonedNode->next=pNode->next;
                pClonedNode=pClonedNode->next;
                pNode->next=pClonedNode->next;
                pNode=pNode->next;
            }
                return pClonedNode;
        }
};