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
作为传入参数。
示例 1:
输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]
示例 2:
输入:head = [[1,1],[2,1]]
输出:[[1,1],[2,1]]
示例 3:
输入:head = [[3,null],[3,0],[3,null]]
输出:[[3,null],[3,0],[3,null]]
示例 4:
输入:head = []
输出:[]
解释:给定的链表为空(空指针),因此返回 null。
提示:
0 <= n <= 1000
-10000 <= Node.val <= 10000
Node.random
为空(null)或指向链表中的节点。
解答:
/*
// 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 {
//哈希表+回溯
//时间复杂度O(N),空间复杂度O(N)
HashMap<Node, Node> visitedHash = new HashMap<>();
public Node copyRandomList(Node head) {
if(head == null) return null;
if(this.visitedHash.containsKey(head)){
return this.visitedHash.get(head);
}
Node node = new Node(head.val, null, null);
this.visitedHash.put(head, node);
node.next = this.copyRandomList(head.next);
node.random = this.copyRandomList(head.random);
return node;
}
}
class Solution {
//哈希表+迭代
//时间复杂度O(N),空间复杂度O(N)
HashMap<Node, Node> visited = new HashMap<>();
public Node getClonedNode(Node node){
if(node != null){
if(!this.visited.containsKey(node)){
this.visited.put(node, new Node(node.val, null, null));
}
return this.visited.get(node);
}
return null;
}
public Node copyRandomList(Node head) {
if(head == null) return null;
Node oldNode = head;
Node newNode = new Node(oldNode.val);
this.visited.put(oldNode, newNode);
while(oldNode != null){
newNode.random = this.getClonedNode(oldNode.random);
newNode.next = this.getClonedNode(oldNode.next);
oldNode = oldNode.next;
newNode = newNode.next;
}
return this.visited.get(head);
}
}
class Solution {
//复制新节点连接在原链表节点后, 连接random, 拆分链表
//时间复杂度O(N),空间复杂度O(1)
public Node copyRandomList(Node head) {
if(head == null) return null;
Node ptr = head;
while(ptr != null){
Node newNode = new Node(ptr.val);
newNode.next = ptr.next;
ptr.next = newNode;
ptr = newNode.next;
}
ptr = head;
while(ptr != null){
ptr.next.random = (ptr.random != null) ? ptr.random.next : null;
ptr = ptr.next.next;
}
Node ptr_old_list = head;
Node ptr_new_list = head.next;
Node ansHead = head.next;
while(ptr_old_list != null){
ptr_old_list.next = ptr_old_list.next.next;
ptr_new_list.next = (ptr_new_list.next == null) ? null : ptr_new_list.next.next;
ptr_old_list = ptr_old_list.next;
ptr_new_list = ptr_new_list.next;
}
return ansHead;
}
}
参考自:
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/copy-list-with-random-pointer/solution/fu-zhi-dai-sui-ji-zhi-zhen-de-lian-biao-by-leetcod/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。