剑指Offer系列—— 35.复杂链表的复制

最新推荐文章于 2024-09-13 18:30:00 发布

听♞海

最新推荐文章于 2024-09-13 18:30:00 发布

阅读量573

点赞数

分类专栏：剑指Offer系列文章标签：链表数据结构算法

本文链接：https://blog.csdn.net/qq_56687452/article/details/122760863

版权

剑指Offer系列专栏收录该内容

40 篇文章 0 订阅

订阅专栏

本文介绍了如何复制一个复杂链表，其中每个节点包含next和random指针。复制过程分为三步：1) 复制节点并拼接到原链表后；2) 处理复制节点的random指针；3) 抽取新的复制链表并恢复原链表。详细思路和代码实现也在文中给出。

摘要由CSDN通过智能技术生成

题目

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

思路

我们分三步做。

第一步，复制每一个节点，拼接到原链表中节点的后一个位置。例如，如下有ABCDE五个节点，我们复制A为A1拼接到A后，使A指向A1，A1指向B，以此列推。第一步完成后，举个例子，如第二张图，将由上图变为下图

在这里插入图片描述

在这里插入图片描述第二步，处理random指针

第三步，抽出新链表，同时也可以把原链表恢复原样

代码

/*
// 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) {
    	//第一步
        for (auto p = head; p; p = p->next->next) {
            auto q = new Node(p->val);
            q->next = p->next;
            p->next = q;
        }
        //第二步
        for (auto p = head; p; p = p->next->next) {
            if (p->random) {
                p->next->random = p->random->next;
            }
        }
        //第三步
        auto dummy = new Node(-1), cur = dummy;
        for (auto p = head; p; p = p->next) {
            cur->next = p->next;
            cur = p->next;
            p->next = cur->next;
        }
        return dummy->next;
    }
};