JZ25_复杂链表的复制

知识点：链表
题目链接

题目描述
输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针random指向一个随机节点），请对此链表进行深拷贝，并返回拷贝后的头结点。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空

解题思路

1. 复制next 如原来是A->B->C->D 变成A->A’->B->B’->C->C’->D->D’
2. 复制random 添加random指针 若原来A->random = B 要让复制的结点也添加上 A->next->random = A->random->next 也就是 A’->random = B’ 否则就random指向空
3. 拆分链表 新头结点=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的一天！