复杂链表的复制
剑指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; } };