题目:输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
解决本题最大的困难就是结点随机指针的赋值,因为并不清楚复制后的随机指针所指向的位置,因此需要确定随机指针位置,才能完成复制
*解题思路:
*1、遍历链表,复制每个结点,如复制结点A得到A1,将结点A1插到结点A后面;
*2、重新遍历链表,复制老结点的随机指针给新结点,如A1.random = A.random.next;
*3、拆分链表,将链表拆分为原链表和复制后的链表
class Solution {
public:
void cloneNodes(RandomListNode *pHead){//step1:完成值和next指针的拷贝
RandomListNode *pNode=pHead;
while(pNode!=nullptr){
//每一次复制都将A的值和next指针拷贝给A',而A’的随即指针先置空
RandomListNode *pClone=new RandomListNode(-1);
//这一步刚开始用RandomListNode *pClone=nullptr;这种方式报错了
pClone->label=pNode->label;
pClone->next=pNode->next;
pClone->random=nullptr;
//A的next指向自己的克隆A',并且后移A至A'的next,也就是原始链表中A的next
pNode->next=pClone;
pNode=pClone->next;
}
}
void connectRandomClone(RandomListNode *pHead){//step2:完成随机指针的拷贝
RandomListNode *pNode=pHead;
while(pNode!=nullptr){
RandomListNode *pClone=pNode->next;
if(pNode->random!=nullptr){//链表结构已经是A->A'->B->B'...这样的结构,因此对于随机指针不为空的结点,克隆结点的随机指针就是原始结点随机指针的下一个结点
pClone->random=pNode->random->next;
}
pNode=pClone->next;
}
}
RandomListNode *reConnectNodes(RandomListNode *pHead){//step3:拆分链表,得到复制到的链表(奇数是原始链表结点,偶数位置是复制链表结点)
RandomListNode *pNode=pHead;
RandomListNode *pCloneHead=nullptr;//头节点记录链表起始位置
RandomListNode *pCloneNode=nullptr;
if(pNode!=nullptr){//首先得到复制链表的头节点
pCloneHead=pCloneNode=pNode->next;
pNode->next=pCloneNode->next;//pNode断开原来链接,后移到pClone的下一个位置
pNode=pNode->next;
}
while(pNode!=nullptr){
pCloneNode->next=pNode->next;//pClone指向pNode的下一个位置
pCloneNode=pCloneNode->next;//后移到下一个位置
pNode->next=pCloneNode->next;;
pNode=pNode->next;
}
return pCloneHead;
}
RandomListNode* Clone(RandomListNode* pHead)
{
cloneNodes(pHead);
connectRandomClone(pHead);
return reConnectNodes(pHead);
}
};