输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
由于是深copy,随机节点指向的对象也应该是新的节点。
给出两种解法
第一种,采用HashMap记录新旧节点
代码如下:
public RandomListNode Clone(RandomListNode pHead)
{
HashMap<RandomListNode, RandomListNode> map = new HashMap<>();
RandomListNode p = pHead;
while(p!=null){
map.put(p, new RandomListNode(p.label));
p = p.next;
}
p = pHead;
while(p!=null){
map.get(p).random = map.get(p.random);
map.get(p).next = map.get(p.next);
p = p.next;
}
return map.get(pHead);
}
第二种方法,不需要额外的空间开销,时间复杂度也为
O(n)
具体
原来的链表为1->2->3
构建新链表的时候,将copy的节点放在原节点后面,即
1->1’->2->2’->3->3’
这样一来,newNode.random = orgNode.random.next
代码如下:
public RandomListNode Clone(RandomListNode pHead)
{
if(pHead ==null) return null;
RandomListNode p = pHead;
RandomListNode next = null;
while(p!=null){
next = p.next;
p.next = new RandomListNode(p.label);
p.next.next = next;
p = next;
}
p = pHead;
RandomListNode curCopy = null;
while(p!=null){
next = p.next.next;
curCopy = p.next;
curCopy.random = p.random !=null? p.random.next:null;
p = next;
}
RandomListNode res = pHead.next;
p = pHead;
while(p!=null){
next = p.next.next;
curCopy = p.next;
p.next = next;
curCopy.next = next!=null?next.next:null;
p = next;
}
return res;
}