题目描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，
另一个特殊指针指向任意一个节点），返回结果为复制后复杂链表的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) {
        if(pHead == nullptr)
            return nullptr;
        auto cur = pHead;
        //拷贝每个节点
        while(cur != nullptr) {
            auto node = new RandomListNode(cur->label);
            node->next = cur->next;
            cur->next = node;
            cur = node->next;
        }
        cur = pHead;
        //对新拷贝的节点变更random指针
        while(cur != nullptr && cur->next != nullptr) {
            auto tmp = cur->next;
            if(cur->random != nullptr)
                tmp->random = cur->random->next;
            cur = tmp->next;
        }
        cur = pHead;
        auto cloneHead = cur->next;
        //拆分
        while(cur != nullptr && cur->next != nullptr) {
            auto tmp = cur->next;
            cur->next = tmp->next;
            cur = tmp;
        }
        return cloneHead;
    }
};