剑指offer之复杂链表的复制

1.题目描述

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

节点的结构体：

struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }

2.分析

• 第一步：在原来的链表基础上，插入新的节点在原有节点的后面，如：a->b->c->NULL，我们完成了这一步之后，链表应该是：a->a’->b->b’->c->c’->null
• 第二步，我们复制节点的random指针
• 第三步，把链表拆分成两个链表

3.代码

RandomListNode* Clone(RandomListNode* pHead)
{
    if(pHead == nullptr)
        return nullptr;
    RandomListNode* pCur = pHead;
    //复制节点
    while(pCur)
    {
        RandomListNode* pNode = new RandomListNode(pCur->label);
        pNode->next = pCur->next;
        pCur->next = pNode;
        pCur = pNode->next;
    }
    pCur = pHead;
    //复制random
    while(pCur)
    {
        RandomListNode* pRandom = pCur->random;
        RandomListNode* pNewNode = pCur->next;
        if(pRandom)
        {
            pNewNode->random = pRandom->next;
        }
        pCur = pNewNode->next;
    }
    pCur = pHead;
    RandomListNode* pNewHead = pCur->next;
    //拆分链表
    while(pCur)
    {
        RandomListNode* pNext = pCur->next;
        pCur->next = pNext->next;
        if(pNext->next)
            pNext->next = pNext->next->next;
        else
        {
            pNext->next = nullptr;
            break;
        }
        pCur = pCur->next ;
    }
    return pNewHead;
}