剑指刷题之复杂链表复制

- 题目要求

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

- 题目分析
采用分治法解决该问题。一共分三步
第一步，复制链表，插入的节点random指针均为空
第二步，根据各节点位置关系，更新random指针
第三步，将复制的链表split得到结果。

- 代码

//节点定义
struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }
};
class Solution {
public:
    RandomListNode* Clone(RandomListNode* pHead)
    {
        CopyList(pHead);
        LinkList(pHead);
        return SplitList(pHead);
    }
    
    RandomListNode* CopyList(RandomListNode* pHead)
    {
        RandomListNode* curNode = pHead;
        while(curNode != NULL)
        {
            RandomListNode* copyNode = new RandomListNode(curNode->label);//A'
            copyNode->next = curNode->next;//A'->B
            curNode->next = copyNode;//A->A'->B
            curNode = copyNode->next;//copy B to B'
            
        }
        return pHead;
    }
    
    RandomListNode* LinkList(RandomListNode* pHead)
    {
        RandomListNode* curNode = pHead;
        while (curNode != NULL)
        {
            RandomListNode* copyNode = curNode->next;
            if(curNode->random)
            {
                copyNode->random = curNode->random->next;//B->D->D',B->B'->D'
            }
            curNode = copyNode->next;
        }
        return pHead;
    }
    
    RandomListNode* SplitList(RandomListNode* pHead)
    {
        //A->A'->B->B'
        if(pHead == NULL)
        {
            return NULL;
        }
        RandomListNode* curNode = pHead;//A
        RandomListNode* pCopyHead = pHead->next;//A'
        RandomListNode* curCopyNode = pCopyHead;//A'
        while (curNode != NULL)
        {
            curNode->next = curCopyNode->next;//A->B
            if(curCopyNode->next != NULL)
            {
                curCopyNode->next = curCopyNode->next->next;//B->B'
            }
            curNode = curNode->next;
            curCopyNode = curCopyNode->next;
        }
        return pCopyHead;
    }
};

结构体指针初始化：

RandomListNode* copyNode = new RandomListNode(curNode->label);//A'