leetcode138: Copy List with Random Pointer

A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.

Return a deep copy of the list.

/**
 * Definition for singly-linked list with a random pointer.
 * class RandomListNode {
 *     int label;
 *     RandomListNode next, random;
 *     RandomListNode(int x) { this.label = x; }
 * };
 */
//import java.util.HashMap;
public class Solution {
    public RandomListNode copyRandomList(RandomListNode head) {
        if(head==null) return null; 
        //保存各节点random指针位置
        HashMap<RandomListNode,RandomListNode> map=new HashMap<RandomListNode,RandomListNode>();
        RandomListNode pp=head;//保存头结点
        while(head!=null){
            map.put(head, head.random);
            head=head.next;
        }
        //深度复制
        head=pp;//回到头结点
        RandomListNode p=head.next;
        RandomListNode pre=head;//用于连接链表, 先不管新链表的头节点的random指针，等到其他节点指针指好后，在考虑它。

        RandomListNode one=pre;//保存新链表的头结点
        while(p!=null){//从第二个节点开始复制，需要保留前一个指针，你懂的。
            RandomListNode lnode=new RandomListNode(p.label);
            pre.next=lnode;
            lnode.next=null;
            lnode.random=map.get(p);//这块很重要，要知道从哪里获得random指针。
            pre=lnode;//最终pre是最后一个节点
            p=p.next;
        }
        one.random=map.get(head);
        return one;

    }
}