复杂链表的复制

题目描述

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

 

 

 

（图片来自牛客）

 

*解题思路：

1、遍历链表，复制每个结点，如复制结点A得到A1，将结点A1插到结点A后面；

2、重新遍历链表，复制老结点的随机指针的下一个给新结点，如A1.random = A.random.next;

3、拆分链表，将链表拆分为原链表和复制后的链表

 

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)
            return NULL;
        RandomListNode *currentNode = pHead;
//复制
        while(currentNode)
        {
            RandomListNode *node = new RandomListNode(currentNode->label);
            node->next = currentNode->next;
            currentNode ->next = node;
            currentNode = node ->next;
        }
//赋值随机节点
        currentNode = pHead;
        while(currentNode)
        {
            RandomListNode *node = currentNode ->next;
            if(currentNode ->random)
            {
                node ->random = currentNode ->random ->next;
            }
            currentNode = node ->next;
        }
        currentNode = pHead;
        RandomListNode *newHead = pHead ->next;
//拆分
        while(currentNode)
        {
            RandomListNode *node = currentNode ->next;
            currentNode ->next = node ->next;
            node ->next = currentNode ->next == NULL? NULL : currentNode ->next ->next;
            currentNode = currentNode ->next;
        }
        return newHead;
    }
};