【Leet Code】138 随机链表的复制

题目描述

原题连接

给你一个长度为 n 的链表，每个节点包含一个额外增加的随机指针 random ，该指针可以指向链表中的任何节点或空节点。

构造这个链表的 深拷贝。 深拷贝应该正好由 n 个 全新 节点组成，其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点，并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点 。

例如，如果原链表中有 X 和 Y 两个节点，其中 X.random --> Y 。那么在复制链表中对应的两个节点 x 和 y ，同样有 x.random --> y。

返回复制链表的头节点。

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

• val：一个表示 Node.val 的整数。
• random_index：随机指针指向的节点索引（范围从 0 到 n-1）；如果不指向任何节点，则为 null 。

你的代码 只 接受原链表的头节点 head 作为传入参数。

成绩

解题思路

建议直接看源码，然后配合这里的给出的思路图看即可。
因为这里的思路我表述很模糊，而且全画图的话，各节点之间的连接关系又太多，互相交叉，导致图也变得很凌乱，因此这里的过程图就没有绘制了。
源码比较简单，代码量也少，配合图看的话应该能 get 到关键点。

我们先来构造一个符合题目描述的链表（源链表）：

图中 next 指向当前节点的下一个节点；
random 随机指向一个节点。

根据我们构造出的源链表，对它进行深拷贝。

我们先不考虑 random，源链表当成普通链表：

对源链表的深拷贝，只需要遍历它，遍历到一个节点 A1 后，可以拿到 A1 的下一个节点 B1（节点 A1 不是尾节点的情况下），然后创建两个新的节点 A2，B2，加入到新链表中（这个新链表将来是深拷贝的结果）：

那么怎么将 random 维护的随机连接的关系也拷贝到新链表中呢？这里就需要使用 Map 来维护源链表与新链表各节点之间的关系，这里选择 HashMap 实现。

具体做法是在之前遍历拷贝源链表时，将源链表节点当成 key，新链表节点当成 value，存入 HashMap 中，形成如下关系：

目前新链表各节点的 random 还未处理，我们需要再次遍历源链表，并根据 HashMap 中维护的关系来补充新链表节点与 random 的关系，步骤：

1. 遍历源链表
2. 得到一个节点 a1，就根据节点 a1 到 HashMap 中拿这个节点在新链表中对应的节点 a2：Node a2 = map.get(a1)
3. 还需要根据节点 a1 的 random 到 HashMap 中 a1.random在新链表中对应节点 a2Random：Node a2Random = map.get(a1.random)
4. 将 a2 与 a2Random 建立联系：a2.random = a2Random
5. 以此类推，直至遍历完源链表，支持对源链表的深拷贝就完成了

代码实现

/*
// Definition for a Node.
class Node {
    int val;
    Node next;
    Node random;

    public Node(int val) {
        this.val = val;
        this.next = null;
        this.random = null;
    }
}
*/

class Solution {
    public Node copyRandomList(Node head) {
        Node currentNode = head;
        Map<Node, Node> map = new HashMap<>();
        while (currentNode != null) {
            map.put(currentNode, new Node(currentNode.val));
            currentNode = currentNode.next;
        }
        currentNode = head;
        while (currentNode != null) {
            map.get(currentNode).next = map.get(currentNode.next);
            map.get(currentNode).random = map.get(currentNode.random);
            currentNode = currentNode.next;
        }
        return map.get(head);
    }
}