请实现 copyRandomList
函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next
指针指向下一个节点,还有一个 random
指针指向链表中的任意节点或者 null
。
示例
题目分析
这个复杂链表相较于普通链表而言就是多了一个随机指针指向链表中的任意节点或null,那么我们应该向清楚怎么复制一个普通链表,其次在考虑怎么复制一个复杂链表。
普通链表的复制
- 首先创建一个新的链表头节点,newHead
- 创建两个指针分别为current(当前节点),newCurrent(新链表的当前节点)
- 使用while循环遍历原链表,直到current指向最后一个节点
- 在while循环中创建一个新节点(newNode),通过将newCurrent.next指向newNode,同时将newCurrent更新为newNode
- 重复直至遍历完成
- 返回新链表的头节点
代码如下
class Node {
int data;
Node next;
public Node(int data) {
this.data = data;
this.next = null;
}
}
public class LinkedListCopy {
public Node copyLinkedList(Node head) {
if (head == null) {
return null;
}
Node newHead = null; //新链表的头节点
Node current = head; //当前节点
Node newCurrent = null; //新链表的头节点
while (current != null) {
Node newNode = new Node(current.data);//复制旧链表的值
if (newHead == null) {//若新链表头节点为空先为新链表头节点赋值
newHead = newNode;
newCurrent = newNode;
} else {
newCurrent.next = newNode;
newCurrent = newCurrent.next;//将当前节点移向最后一个节点
}
current = current.next;//旧链表的节点继续向后移动
}
return newHead;//返回新链表的头节点
}
}
复杂链表的复制
- 复杂链表相较于普通链表是多了一个随机指针的,因此我们不能使用双指针的方法进行复制
- 考虑使用哈希表的方法来实现
- 遍历旧链表,使用哈希表(map)存储旧链表节点和新链表节点的对应关系
- 再次遍历,将新链表的next指针与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;
}
HashMap<Node,Node> map = new HashMap<>();
Node current = head;
while(current != null){//第一次遍历,存储旧链表与新链表的对应关系
Node node = new Node(current.val);
map.put(current,node);
current = current.next;
}
current = head;//将节点重新指向旧链表的头节点
while(current != null){//第二次遍历,将新链表的的next与random进行链接
map.get(current).next = map.get(current.next);
map.get(current).random = map.get(current.random);
current = current.next;
}
return map.get(head);//返回新链表的头节点
}
}