题目:
一种特殊的链表节点类描述如下:
class Node{
public int value;
public Node next;
public Node rand;
public Node(int data){
this.value=data;
}
}
Node类中的value是节点值,next指针和正常链表中next指针的意义一样,rand指针可能指向任意一个Node,也可以指向null。
给定一个由Node节点类型组成的无环单链表的头结点head,请实现一个函数完成这个链表结构的复制,并返回新链表的头结点。例如:链表1->2->3->null,假设1的rand指针指向3,2的rand指正指向null,3的rand指针指向1。复制后的链表应该也是这种结构。
解法一:HashMap解决
1。遍历链表一次,设置key为Node,value为复制的Node的value,没有设置next和rand。
2。第二次遍历设置next和rand。
3。返回头
/**
* 普通解法利用HashMap
* @param head
* @return
*/
public Node copyListWithRand1(Node head){
HashMap<Node,Node> map=new HashMap<Node, Node>();
Node cur=head;
//遍历第一次链表
while (cur!=null){
map.put(cur,new Node(cur.value));
cur=cur.next;
}
cur=head;
//遍历第二次,设置next和rand
while (cur!=null){
map.get(cur).next=map.get(cur.next);
map.get(cur).rand=map.get(cur.rand);
cur=cur.next;
}
//返回头
return map.get(head);
}
该解法时间复杂度为O(n),空间复杂度为O(n)。
解法二:刚才我们利用了额外空间来计算,接下来我们不利用额外空间,并且时间复杂度为O(n)的算法。
1。首先我们把链表的形式由1-2-3-null变为1-1’-2-2’-3-3’-null,意思就是把我们复制的链表插在我们的中间。
2。复制rand。
3。链表分离。
/**
* 进阶解法不利用HashMap
* @param head
* @return
*/
public Node copyListWithRand2(Node head){
if (head==null){
return null;
}
Node cur=head;
Node next=null;
//遍历一次,复制并链接每一个节点
while (cur!=null){
next=cur.next;
cur.next=new Node(cur.value);
cur.next.next=next;
cur=next;
}
cur=head;
Node curCopy=null;
//设置复制节点的rand指针
while (cur!=null){
next=cur.next.next;
curCopy=cur.next;
curCopy.rand=cur.rand!=null?cur.rand.next:null;
cur=next;
}
Node res=head.next;
cur=head;
//拆分
while (cur!=null){
next=cur.next.next;
curCopy=cur.next;
cur.next=next;
curCopy.next=next!=null?next.next:null;
cur=next;
}
return res;
}