A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.
Return a deep copy of the list.
/**
* Definition for singly-linked list with a random pointer.
* class RandomListNode {
* int label;
* RandomListNode next, random;
* RandomListNode(int x) { this.label = x; }
* };
*/
// hashmap version O(n) space
public class Solution {
public RandomListNode copyRandomList(RandomListNode head) {
if (head == null) {
return null;
}
Map<RandomListNode, RandomListNode> map = new HashMap<>();
RandomListNode dummy = new RandomListNode(0);
RandomListNode pre = dummy;
while(head != null) {
if (!map.containsKey(head)) {
RandomListNode newNode = new RandomListNode(head.label);
map.put(head, newNode);
}
if(head.random != null) {
if (!map.containsKey(head.random)) {
RandomListNode newRandom = new RandomListNode(head.random.label);
map.put(head.random, newRandom);
}
}
pre.next = map.get(head);
if (head.random != null) {
pre.next.random = map.get(head.random);
}
pre = pre.next;
head = head.next;
}
return dummy.next;
}
}
/*
第一遍扫的时候巧妙运用next指针,
开始数组是1->2->3->4 。
然后扫描过程中 先建立copy节点 1->1`->2->2`->3->3`->4->4`,
然后第二遍copy的时候去建立边的copy, 拆分节点,
一边扫描一边拆成两个链表,这里用到两个dummy node。
第一个链表变回 1->2->3 , 然后第二变成 1`->2`->3`
*/
//No HashMap version
public class Solution {
private void copyNext(RandomListNode head) {
while (head != null) {
RandomListNode newNode = new RandomListNode(head.label);
newNode.random = head.random;
newNode.next = head.next;
head.next = newNode;
head = head.next.next;
}
}
private void copyRandom(RandomListNode head) {
while (head != null) {
if (head.next.random != null) {
head.next.random = head.random.next;
}
head = head.next.next;
}
}
private RandomListNode splitList(RandomListNode head) {
RandomListNode newHead = head.next;
while (head != null) {
RandomListNode temp = head.next;
head.next = temp.next;
head = head.next;
if (temp.next != null) {
temp.next = temp.next.next;
}
}
return newHead;
}
public RandomListNode copyRandomList(RandomListNode head) {
if (head == null) {
return null;
}
copyNext(head);
copyRandom(head);
return splitList(head);
}
}