leetcode138: Copy List with Random Pointer

shuaishuai3409發表於2016-05-12

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;

    }
}

相關文章