题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),
返回结果为复制后复杂链表的head
使用哈希表
使用一个哈希表,来存储原节点和拷贝节点
public class Solution {
public RandomListNode Clone(RandomListNode pHead){
HashMap<RandomListNode, RandomListNode> map = new HashMap<>();
RandomListNode cur = pHead;
while(cur!=null){ //遍历链表,并拷贝,然后都存到哈希表中
map.put(cur,new RandomListNode(cur.label));
cur=cur.next;
}
cur=pHead;
while(cur!=null){
//连接拷贝的当前节点的next -> 拷贝的next
map.get(cur).next = map.get(cur.next);
//连接拷贝的当前节点的rand -> 拷贝的rand
map.get(cur).random = map.get(cur.random);
cur=cur.next;
}
return map.get(pHead);//返回拷贝的第一个节点
}
}
先复制再拆分
1、遍历链表,复制每个结点,如复制结点A得到A1,将结点A1插到结点A后面;B1查到B后面…
2、重新遍历链表,复制老结点的随机指针给新结点,如A1.random = A.random.next;
3、拆分链表,将链表拆分为原链表和复制后的链表
public class Solution {
public RandomListNode Clone(RandomListNode pHead){
if(pHead==null) return null;
RandomListNode cur=pHead;
RandomListNode next=null;
while(cur!=null){
next=cur.next; //存储下一个节点
cur.next=new RandomListNode(cur.label); //让1的next指向1',1'指向2,2指向2' ...
cur.next.next=next; //1'指向2
cur=next;
}
cur = pHead;
RandomListNode copy=null;
while(cur!=null){ //每次拿出1个节点和它的复制节点,如1和1',2和2'
next=cur.next.next;
copy=cur.next;
//cur.rand表示原来的随机节点,cur.rand.next表示它的复制
copy.random=cur.random!=null?cur.random.next:null;
cur=next;
}
RandomListNode res=pHead.next;
cur=pHead;
while(cur!=null){ //把这个大链表分离
next=cur.next.next;
copy=cur.next; //cur.next表示是复制的节点
cur.next=next; //cur是原来的节点
copy.next=next!=null?next.next:null;
cur=next;
}
return res;
}
}