JZ25_复杂链表的复制
知识点:链表
题目链接
题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空
解题思路
- 复制next 如原来是A->B->C->D 变成A->A’->B->B’->C->C’->D->D’
- 复制random 添加random指针 若原来A->random = B 要让复制的结点也添加上 A->next->random = A->random->next 也就是 A’->random = B’ 否则就random指向空
- 拆分链表 新头结点=pHead->next 然后就是拆分 和删除一个结点类似 多一步 把删除的结点 再和后面的相连 注意最后是否是空结点
代码
#include "cheader.h"
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 pHead;
RandomListNode* pcur = pHead;
//复制next
while(pcur != nullptr){
RandomListNode *tmp = new RandomListNode(pcur->label);
RandomListNode *pcur_next = pcur->next;
pcur->next = tmp;
tmp->next = pcur_next;
pcur = pcur_next;
}
// 添加random指针
pcur = pHead;
while(pcur != nullptr){
if(pcur->random == nullptr)//如果原来结点没有random 就不加
pcur->next->random = nullptr;
else //如果原来有 指向 原来指向的后一个(复制出来的那个)
pcur->next->random = pcur->random->next;
pcur = pcur->next->next;
}
// 拆分链表
RandomListNode *nHead = pHead->next;
pcur = pHead;
while(pcur != nullptr){
RandomListNode *nnode = pcur->next;
pcur->next = nnode->next;
nnode->next = pcur->next == nullptr? nullptr :pcur->next->next;
pcur = pcur->next;
}
return nHead;
}
};
今天也是爱zz的一天!