学习笔记LeetCode剑指Offer 35. 复杂链表的复制

Posted 2021-11-09 一入机械深似海

题目：

请实现 copyRandomList 函数，复制一个复杂链表。在复杂链表中，每个节点除了有一个 next 指针指向下一个节点，还有一个 random 指针指向链表中的任意节点或者 null。

• 迭代 + 节点拆分法

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;
    
    Node(int _val) {
        val = _val;
        next = NULL;
        random = NULL;
    }
};
*/
class Solution {
public:
    Node* copyRandomList(Node* head) {
        //1. 在每个原结点后复制该结点
        for(auto p = head; p; )//死循环，退出条件为p == nullptr
        {
            //将原结点的val赋给新结点的_val
            Node* newp = new Node(p->val);
            auto temp = p->next;//缓存p的下一个结点地址
            p->next = newp;//将p的下一个结点变成新复制的结点
            newp->next = temp;
            p = temp;//每次新增结点后将p移动到下一个原结点处进行新一轮的复制
        }
        //2. 添加新增的复制结点的random值
        for(auto p1 = head; p1; p1 = p1->next->next)//每次循环往后移动两个结点
        {
            if(p1->random){ // !!!不要把这个判断条件丢了
                //假设原来A的random指向C，那么此处让A'的random指向C'
                p1->next->random = p1->random->next;
            }
        }
        //3. 拆分原结点和新结点
        //新建一个哑结点，简化代码，省略当函数的入口参数为空时的判断
        Node* dummy = new Node(-1);//val为-1的哑结点，用指针dummy接收该结点地址
        auto cur = dummy;//dummy不动，用于最后返回其next的head；复制一个起始指向哑结点的指针cur，让其移动

        for(auto p2 = head; p2; p2 = p2->next)//将指针p2从A移动到B
        {
            //A  A' B  B'...
            cur->next = p2->next;//将哑结点cur与A'相连
            cur = cur->next;//将指针cur移动到A'
            p2->next = p2->next->next;//A与B相连
        }
        return dummy->next;
    }
};