[剑指offer]-复杂链表的复制

题目描述

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

思路：(在不借助辅助空间的情况下，实现O(n)的时间效率)

分为三步走：

代码如下：

/*
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 == NULL)
            return NULL;
        //第一步：遍历链表，复制每个结点，如复制结点A得到A1，将结点A1插到结点A后面；
        RandomListNode* curNode = pHead;
        while(curNode != NULL){
            RandomListNode* pCloned = new RandomListNode(0);
            pCloned->label = curNode->label;
            pCloned->next = curNode->next;
            pCloned->random = NULL;
            curNode->next = pCloned;
            curNode = pCloned->next;
        }
        //第二步：重新遍历链表，复制老结点的随机指针给新结点，如A1.random = A.random.next;
        curNode = pHead;
        while(curNode != NULL){
            RandomListNode* pCloned = curNode->next;
            if(curNode->random != NULL){  //！！！！很关键的一个判断，不然就出现野指针
                pCloned->random = curNode->random->next;
            }
            curNode = pCloned->next;
        }
        //第三步：拆分链表，将链表拆分为原链表和复制后的链表
        curNode = pHead;
        RandomListNode* pCloneHead = pHead->next;
        while(curNode != NULL){
            RandomListNode* pCloneNode = curNode->next;  //临时存放当前节点的下一个指针
            curNode->next = curNode->next->next;  //原始链表，隔一个节点连接一个
            //复制后链表，隔一个节点连接一个，但是要考虑该节点是最后一个的情况
            pCloneNode->next = pCloneNode->next == NULL?NULL:pCloneNode->next->next;
            curNode = curNode->next;  //移到下一个原始节点
        }
        return pCloneHead;
    }
};