输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路:刚开始想到的是很直接的想法,先按照next完全clone一遍,并放到一个数组里面,然后对原链表做处理,用Map储存每个Node对应的index,然后在clone出来的链表里面找到random对应的index
书上有个更简单的不需要额外空间的解决办法,就是把clone的链表放在原始链表的后面,然后更新random,因为维持了原始的顺序,所以random.next == next.random,最后再把链表split成2个就OK了
这种利用原始链表信息的解法还是有一定的启发的
/*
public class RandomListNode {
int label;
RandomListNode next = null;
RandomListNode random = null;
RandomListNode(int label) {
this.label = label;
}
}
*/
import java.util.*;
public class Solution {
public RandomListNode Clone(RandomListNode pHead)
{
if(pHead == null) return null;
cloneOneByOne(pHead);
cloneRandom(pHead);
return split(pHead);
}
public void cloneOneByOne(RandomListNode pHead)
{
while(pHead != null) {
RandomListNode newNode = new RandomListNode(pHead.label);
newNode.next = pHead.next;
pHead.next = newNode;
pHead = newNode.next;
}
}
public void cloneRandom(RandomListNode pHead)
{
while(pHead != null) {
if(pHead.random != null)
pHead.next.random = pHead.random.next;
pHead = pHead.next.next;
}
}
public RandomListNode split(RandomListNode pHead)
{
RandomListNode tmp = pHead.next;
RandomListNode root1 = pHead, root2 = pHead.next;
while(pHead.next.next != null) {
pHead.next = pHead.next.next;
pHead = pHead.next;
tmp.next = tmp.next.next;
tmp = tmp.next;
}
pHead.next = null;
return root2;
}
}