复杂链表的复制

[TOC]
今天在牛客网上做了一道题，很有意思就分享出来了。

题目：

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点），返回结果为复制后复杂链表的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)
    {

    }
};

初始思路（错误思路）：

初看题目，稍稍一思索就想着，这题不难，分两步即可，第一步：把原链表节点复制一遍；第二步：在把新链表的的特殊指针一置就OK了。貌似思路很清晰，我就开始撸代码，撸到置特殊指针我就发现不对了，这是个单链表，每次置特殊指针都需要遍历一遍链表。这复杂度就妥妥O(N^2) 了。我还是坚持写完了代码，在线测试不过，复杂度过大。无奈的我只能想其他办法了。

新思路（正确思路）：

第一步：
 把原链表的每个节点进行复制，并且把新链表的节点链在原链表的后面。
第二步：
 置新节点的random指针
第三步：
 把新节点连起来，并且与原链表的节点断开

代码展示：

struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
    label(x), next(NULL), random(NULL) {
    }
RandomListNode* Clone(RandomListNode* pHead)
{
    RandomListNode* cur = pHead;
    RandomListNode* ret;
    if(!cur)
        return NULL;
    while (cur) //把原链表节点复制一遍
    {
        RandomListNode* tmp = new RandomListNode(cur->label);
        tmp->next = cur->next;
        cur->next = tmp;
        cur = tmp->next;
    }

    cur = pHead;
    ret = pHead->next;
    RandomListNode* tmp = ret;
    while (cur) //设置新链表的random值
    {
        if(cur->random)
           ret->random = cur->random->next;
        cur = ret->next;
        ret = cur->next;
    }

    cur = pHead;
    ret = pHead->next;
    tmp = ret;

    cur->next = tmp->next;
    cur = cur->next;
    while(cur) //断掉旧的链表
    {
        tmp->next = cur->next;
        tmp = tmp->next;

        RandomListNode* newNode = cur->next;
        cur->next = newNode->next;
        cur = cur->next;

    }
    return ret;
}

这样就把O(N^2)的时间复杂度降到了O(N)。