剑指offer:复制复杂链表

题目叙述:

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点），返回结果为复制后复杂链表的head。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

分析:

主要是分为两步:

• 先不考虑被复制链表的random数据域正常扫描链表，进行复制，复制的过程中建立要创建的新结点与被复制结点的哈希映射
• 再次扫描被复制的链表，通过访问被复制链表的random数据域根据哈希映射即可得到新链表的random数据域，从而完成复杂链表的复制。

代码：

import java.util.HashMap;
import java.util.Map;
public class Solution {
    public RandomListNode Clone(RandomListNode pHead)
    {
        if (pHead == null) return null;
        RandomListNode newhead =  new RandomListNode(pHead.label);
        RandomListNode pre = newhead;
        //为方便每一个复制的结点快速找到它的next结点，考虑使用哈希表进行映射
        Map<RandomListNode, RandomListNode> map = new HashMap<RandomListNode,RandomListNode>();
        map.put(pHead,newhead); 
        RandomListNode p = pHead.next;
        while(p!=null){
           RandomListNode newnode =  new RandomListNode(p.label);
           pre.next = newnode;
           pre = newnode;    
           map.put(p,newnode); 
           p = p.next;
        }
        p = pHead;
        RandomListNode q = newhead;
        while(p!=null){
           q.random = map.get(p.random);
           p = p.next;
           q = q.next;
        }
        return newhead;
    }
}