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

Posted 2020-10-16 zmj97

题目描述

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

---

 

首先在原链表每个结点后边插入一个结点，其label等于此结点的label，

然后把每一个复制结点的随机指针指向它复制的那个结点的随机指针指向的结点的后一个（即其复制结点），

然后将原结点和复制结点分离开即可。

 

/*
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;
        // 复制结点
        RandomListNode* tmp = pHead;
        while (tmp != NULL) {
            RandomListNode* tmpCopy = new RandomListNode(tmp->label);
            tmpCopy->next = tmp->next;
            tmp->next = tmpCopy;
            tmp = tmpCopy->next;
        }
        // 添加随机指针
        tmp = pHead;
        while (tmp != NULL) {
            if (tmp->random) tmp->next->random = tmp->random->next;
            tmp = tmp->next->next;
        }
        // 分离结点
        RandomListNode* re = pHead->next;
        tmp = pHead;
        while (tmp != NULL) {
            RandomListNode *tmpCopy = tmp->next;
            tmp->next = tmpCopy->next;
            tmp = tmp->next;
            if (tmp) tmpCopy->next = tmp->next;
        }
        return re;
    }
};