剑指offer——复杂链表的复制*

题目描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点），返回结果为复制后复杂链表的head。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

 

解题思路：

分成3步：

1. 在每个节点后面复制这个节点，插入链表中 类似A a B b C c D d,小写为复制的  
2. 复制random的指向
3. 拆分

需要注意的就是最后一步，很容易空指针错误，错了好多次。。。

 

/*
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;
        
        //在每个节点后面复制这个节点，插入链表中 类似A a B b C c D d,小写为复制的  
        RandomListNode* pNode=pHead;
        while(pNode!=nullptr){
            RandomListNode* pCloned=new RandomListNode(pNode->label);
            pCloned->next=pNode->next;
            pCloned->random=nullptr;
            pNode->next=pCloned;
            pNode=pCloned->next;
        }
        
        //复制random的指向
        pNode=pHead;
        while(pNode!=nullptr){
            RandomListNode* pCloneNode=pNode->next;
            if(pNode->random!=nullptr) pCloneNode->random=pNode->random->next;
            pNode=pCloneNode->next;
        }
        
        //拆分
        RandomListNode* pClonedHead=pHead->next;
        RandomListNode* pClonedNode=pClonedHead;
        pNode=pHead;
        pNode->next=pClonedNode->next;
        pNode=pNode->next;
        while(pNode!=nullptr){
            pClonedNode->next=pNode->next;
            pClonedNode=pClonedNode->next;
            pNode->next=pClonedNode->next;
            pNode=pNode->next;
        }
       
        return pClonedHead;
    }
};