复杂链表的复制

题目描述

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

代码

/*
struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }
};
*/
class Solution {
public:
    //先进行next拷贝,在进行random拷贝,时间复杂度为o(n2)
    RandomListNode* Clone(RandomListNode* pHead)
    {
        RandomListNode* newHead = cloneNext(pHead);//next链表拷贝
        
       	RandomListNode* cur = newHead, *curp = pHead;
        while (cur != NULL) {
            if (curp->random != NULL) {
                cur->random = cloneRandom(pHead, curp->random);//单个random指针拷贝
            }
            curp = curp->next;
            cur = cur->next;
        }
        return newHead;
    }
    
    RandomListNode* cloneNext(RandomListNode* pHead) 
    {
        if (pHead == NULL) {
            return NULL;
        }
        RandomListNode* newNode = new RandomListNode(pHead->label);
        newNode->next = cloneNext(pHead->next);
        return newNode;
    }
    
    RandomListNode* cloneRandom(RandomListNode* pHead, RandomListNode* random) {
        if (pHead == random) {
            return new RandomListNode(pHead->label); 
        }
        return cloneRandom(pHead->next, random);
    }
    
};