Leetcode 138. 复制带随机指针的链表

JehanRio

已于 2023-10-01 09:29:40 修改

阅读量266

点赞数 1

分类专栏： Leetcode专栏文章标签：链表 leetcode 数据结构

于 2023-05-15 10:04:49 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/130678441

版权

Leetcode专栏专栏收录该内容

278 篇文章 2 订阅

订阅专栏

题目

在这里插入图片描述

138. 复制带随机指针的链表

代码（首刷看解析）

/*
// 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) {
        // 防止重复创建，先全部创建好
        unordered_map<Node*, Node*> mp;
        for(auto p = head; p != nullptr; p = p->next) {
            if(!mp.count(p)) {
                mp[p] = new Node(p->val);
            }
        }
        auto ans = head;
        while(head) {
            if(head->next) {
                mp[head]->next =  mp[head->next];
            }
            if(head->random) {
                mp[head]->random = mp[head->random];
            }
            head = head->next;
        }
        return mp[ans];
    }
};

代码（9.7 二刷看解析）

class Solution {
public:
    Node* copyRandomList(Node* head) {
        unordered_map<Node*, Node*> mp; // 指向随机节点
        auto p = head;
        while(p) {
            // if(!mp.count(p))
                mp[p] = new Node(p->val);
            p = p->next;
        }
        auto ans = head;
        while(head) {
            if(head->next) 
                mp[head]->next = mp[head->next];
            if(head->random)
                mp[head]->random = mp[head->random];
            head = head->next;
        }
        return mp[ans];
    }
};

代码（10.1 三刷看解析）

class Solution {
public:
    Node* copyRandomList(Node* head) {
        unordered_map<Node*, Node*> mp;
        auto p = head;
        while(p) {
            if(!mp.count(p)) {
                mp[p] = new Node(p->val);
            }
            p = p->next;
        }
        auto q = head;
        while(q) {
            if(q->next)
                mp[q]->next = mp[q->next];
            if(q->random)
                mp[q]->random = mp[q->random];
            q = q->next;
        }
        return mp[head];
    }
};