题目描述
给你一个长度为 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
的关系,步骤:
- 遍历源链表
- 得到一个节点
a1
,就根据节点a1
到HashMap
中拿这个节点在新链表中对应的节点a2
:Node a2 = map.get(a1)
- 还需要根据节点
a1
的random
到HashMap
中a1.random
在新链表中对应节点a2Random
:Node a2Random = map.get(a1.random)
- 将
a2
与a2Random
建立联系:a2.random = a2Random
- 以此类推,直至遍历完源链表,支持对源链表的深拷贝就完成了
代码实现
/*
// 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);
}
}