复杂链表的复制

Posted 2020-11-27 xiaokang01

剑指offer25复制链表的复制

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点），返回结果为复制后复杂链表的head。

（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

 

先说明白什么是复杂链表：

就是A结点中有一个结点值，和两个指针，一个指向下一个结点，一个指向随机结点

 

 我们要对其进行复制，

最明显的思路就是，先对next和label进行复制，然后对当前结点的random进行遍历

从0开始遍历，但是这样复杂度会是O(n^2);

不考虑，

思路二：

// 第一步：将复杂链表的labe和next复制一份，就是复制节点A得到A1，将A1插入到A的后面
// 第二步：同步复制出来的节点的random，（当前的random就是原来节点的random的下一个节点）
// A1->random = A->random->next;
// 第三步，拆分链表（类似链表的删除结点操作）

看图：

第一步：

 

 第二步：

 

 第三步：

 

 

代码：

//
// Created by LK on 2020/3/27.
//
/*
struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }
};
*/

// 第一步：将复杂链表的labe和next复制一份，就是复制节点A得到A1，将A1插入到A的后面
// 第二步：同步复制出来的节点的random，（当前的random就是原来节点的random的下一个节点）
// A1->random = A->random->next;
// 第三步，拆分链表（类似链表的删除结点操作）
class Solution {
public:
    RandomListNode* Clone(RandomListNode* pHead)
    {
        if(pHead == nullptr)
            return nullptr;
        RandomListNode *pCur = pHead;  // 指向当前结点
        while (pCur != nullptr)
        {
            // 复杂结点信息
            RandomListNode *temp = new RandomListNode(pCur->label);
            temp->next = pCur->next;

            // 将temp结点插入A后面
            pCur->next = temp;

            pCur = temp->next; // 自增
        }


        // 此时已经将链表复制了一份，但是random还没有设置，现在开始设置
        pCur = pHead;
        while(pCur != nullptr)
        {
            RandomListNode *temp = pCur->next;  // temp 指向A1

            if(pCur->random != nullptr)
                 temp->random = pCur->random->next;  // 设置A1的random

            pCur = temp->next;  // 自增
        }

        // 此时复杂链表已经全部复制了，只需要将链表拆开（把A和A1分离）
        pCur = pHead;
        RandomListNode *pClone = pHead->next;
        RandomListNode *temp = nullptr;  // 保存要（A1结点）删除的结点

        while(pCur->next)
        {
            temp = pCur->next;
            pCur->next = temp->next;

            pCur = temp; // 自增  因为上面循环是pCur->next 所以是 是指向temp
        }
        return pClone;
    }
};