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

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

给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。

要求返回这个链表的 深拷贝。

我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示:

  • val:一个表示 Node.val 的整数。
  • random_index:随机指针指向的节点索引(范围从 0 到 n-1);如果不指向任何节点,则为 null 。
    LeetCode - 138. 复制带随机指针的链表_第1张图片
    LeetCode - 138. 复制带随机指针的链表_第2张图片
    解题思路: 这道题是《剑指offer》上的原题。解这题的关键点是如何计算random的节点。书上的解法和此题的解法均采用维护一个源节点到目标节点的映射,在遍历源链表时,通过它的random节点即可得到目标节点的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) {
        if (!head) return NULL;
        Node *dummy = new Node(-1), *cur = dummy;
        Node *p = head;
        unordered_map<Node*, Node*> mp;
        while (p) {
            cur->next = new Node(p->val);
            mp[p] = cur->next;
            cur = cur->next;
            p = p->next;
        }
        p = head;
        cur = dummy->next;
        while (p) {
            cur->random = mp[p->random];
            cur = cur->next;
            p = p->next;
        }
        Node *res = dummy->next;
        delete dummy;
        return res;
    }
};

你可能感兴趣的:(LeetCode,leetcode,指针,链表)