1.题目描述
- 复制带随机指针的链表
给你一个长度为 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 作为传入参数。
2.解题思路
-
一、 复制带随机指针的链表(思路)
- 先不考虑 random 的指向,先把旧的链表复制一份,遍历旧就链表的节点,根据节点值 new新节点,尾插到新链表上;
- 再次遍历旧链表。依次考察,求出每个节点的random指向的位置距离头结点的偏移量是几。
-
二、具体实现( 借助 Map 可以有更加方便实现的方案)
-
遍历旧链表,根据旧链表的值创建新节点,以旧节点为 key 新节点为 value,插入到 Map 中
-
再次遍历旧链表,给新节点的 next进行赋值
例如,预期是 让新1.next = 新2。
可以通过map.get(旧1)=> 新1, map.get(旧1.next) => 新2 -
再次遍历旧链表,给新节点的 random进行赋值
例如,新1.random = 新 3。
map.get (旧1) => 新1,map.get(旧1.random) => 新3。
-
3.LeetCode代码实现
/*
// 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) {
// 1.先遍历旧链表,把每个旧节点都创建对应的新节点,并且插入到 Map 中
Map<Node,Node> map = new HashMap<>();
for(Node cur = head; cur != null; cur = cur.next) {
map.put(cur,new Node(cur.val));
}
// 2.再次遍历旧链表,构造 next/ random 的指向
for (Node oldNode = head; oldNode != null; oldNode = oldNode.next) {
//把 next 想回连接
Node newNode = map.get(oldNode);
Node newNodeNext = map.get(oldNode.next);
newNode.next = newNodeNext;
//把 random 相互连接
Node newNodeRandom = map.get(oldNode.random);
newNode.random = newNodeRandom;
}
return map.get(head);
}
}
- 结果