@author: sdubrz
@date: 8/2/2020 3:22:32 PM
难度: 中等
考察内容: 链表
@e-mail: lwyz521604#163.com
题目来自《剑指offer》 电子工业出版社,图片来自LeetCode
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
示例 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。
提示:
- -10000 <= Node.val <= 10000
- Node.random 为空(null)或指向链表中的节点。
- 节点数目不超过 1000 。
一个错误的解法
先根据每个节点的 next 信息,复制所有的节点,然后根据节点的值逐个确定其 random 信息。这是一个时间复杂度为 O(n2)
的方法,并且当节点的取值存在重复时,无法正确确定每个节点的random信息。这是我第一时间想到的方法,提交之后运行结果错误。
/*
// 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) {
if(head==null) {
return null;
}
Node headNode = new Node(head.val);
Node current = headNode;
Node ptrNode = head.next;
while(ptrNode!=null) {
Node temp = new Node(ptrNode.val);
current.next = temp;
current = current.next;
ptrNode = ptrNode.next;
}
current = headNode;
ptrNode = head;
while(current!=null) {
if(ptrNode.random==null) {
ptrNode = ptrNode.next;
current = current.next;
continue;
}
Node temp = headNode;
while(temp.val!=ptrNode.random.val) {
temp = temp.next;
}
current.random = temp;
ptrNode = ptrNode.next;
current = current.next;
}
return headNode;
}
}
第二种方法 借助Map
我想到的第二种思路是将两个链表上相同位置上的节点建立映射关系,可以用HashMap帮助实现:
/*
// 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) {
if(head==null) {
return null;
}
HashMap<Node, Node> map = new HashMap<>();
Node headNode = new Node(head.val);
map.put(head, headNode);
Node current = headNode;
Node ptrNode = head.next;
while(ptrNode!=null) {
Node temp = new Node(ptrNode.val);
current.next = temp;
map.put(ptrNode, temp);
current = current.next;
ptrNode = ptrNode.next;
}
current = headNode;
ptrNode = head;
while(current!=null) {
if(ptrNode.random!=null) {
current.random = map.get(ptrNode.random);
}
current = current.next;
ptrNode = ptrNode.next;
}
return headNode;
}
}
在 LeetCode 系统中提交的结果为
执行结果:通过 显示详情
执行用时:0 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗:39.3 MB, 在所有 Java 提交中击败了88.41%的用户