题目描述:
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
方法一:
1.将每一个节点复制后插入到原节点的后面,新的random域先初始化为null
2.新节点的random为旧节点random指向的下一个节点
3.拆分:将这个链表的新旧节点拆分开,返回新节点
class A{
int data;
A next;
A random;
public A(int data,A next,A random){
this.data=data;
this.next=next;
this.random=random;
}
}
public class RandomCopy {
A head;
public A copyRandomList(){
if(this.head==null){
return null;
}
A cur=this.head;
while(cur!=null){
A node=new A(cur.data,cur.next,null);
A tmp=cur.next;
cur.next=node;
cur=tmp;
}
cur=this.head;
while(cur!=null){
if(cur.random!=null){
cur.next.random=cur.random.next;
}else{
cur.next.random=null;
}
cur=cur.next.next;
}
//拆
cur=this.head;
A newHead=cur.next;
while(cur.next!=null){
A tmp=cur.next;
cur.next=tmp.next;
cur=tmp;
}
return newHead;
}
}
方法二:借助HashMap存储
1.遍历整个链表,将原节点作为键,将复制的新节点作为值存入map中
2.在遍历一遍链表,将新节点的next域和random域根据原节点赋值
class Solution {
public Node copyRandomList(Node head) {
Map<Node,Node> map=new HashMap<>();
Node cur=head;
while(cur!=null){
map.put(cur,new Node(cur.val));
cur=cur.next;
}
cur=head;
while(cur!=null){
map.get(cur).next=map.get(cur.next);
map.get(cur).random=map.get(cur.random);
cur=cur.next;
}
return map.get(head);
}
}