剑指offer 复杂链表的复制

题目描述

原题链接

算法

(线性扫描) $O(n)$

• 复制每个节点，如：复制节点A得到A’，将A’插入节点A后面
• 遍历链表，A’->random = A->random->next;
• 将链表拆分成原链表和复制后的链表

时间复杂度是$O(n)$，空间复杂度是$O(n)$

参考示意图

代码

/*
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) return NULL;
        // A->A'->B->B'->C->C'->NULL
        // 复制每个结点，如复制结点A得到A'，将结点A'插到结点A后面
        auto p = pHead;
        while(p) {
            auto q = new RandomListNode(p->label);
            // 插入A'
            q->next = p->next;
            p->next = q;
            p = p->next->next;
        }
        
        // 复制老结点的随机指针给新结点
        p = pHead;
        while(p) {
            if (p->random) p->next->random = p->random->next; 
            p = p->next->next;
            
        }
        
        // 拆分出原链表和新链表
        auto pCloneHead = pHead->next;
        p = pHead;
        while(p->next) {
            auto tmp = p->next;
            p->next = tmp->next;
            p = tmp;
        }
        
        return pCloneHead;
    }
};

来源：CSDN

作者：Wilson790

链接：https://blog.csdn.net/qq_43827595/article/details/104214242