复杂链表的复制
复杂链表的复制
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
用一个哈希表表示映射关系:键是原节点,值是复制的节点。
整体算法流程是:
- 第一次遍历,复制每个节点和 next 指针,并且保存“原节点-复制节点”的映射关系
- 第二次遍历,通过哈希表获得节点对应的复制节点,更新 random 指针
/*
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;
RandomListNode newHead=new RandomListNode(pHead.label);
RandomListNode saveOld=pHead;
RandomListNode saveNew=newHead;
HashMap<RandomListNode,RandomListNode>mapping=new HashMap<RandomListNode,RandomListNode>();
mapping.put(pHead,newHead);
while(pHead.next!=null){
newHead.next=new RandomListNode(pHead.next.label);
mapping.put(pHead.next,newHead.next);
pHead=pHead.next;
newHead=newHead.next;
}
pHead=saveOld;
newHead=saveNew;
while(pHead!=null){
newHead.random=mapping.get(pHead.random);
pHead=pHead.next;
newHead=newHead.next;
}
return saveNew;
}
}
要点:
- map的插入是put,根据key找value是get
- 注意判断pHead是否为null

浙公网安备 33010602011771号