Hash方法实现复杂链表的复制

题目描述

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

/*
public class RandomListNode {
    int label;
    RandomListNode next = null;
    RandomListNode random = null;

    RandomListNode(int label) {
        this.label = label;
    }
}
*/
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
public class Solution {
    public RandomListNode Clone(RandomListNode pHead)
    {
        if(pHead == null)
            return null;
        
        //开始复制、连接链表
        RandomListNode pNode = pHead;
        RandomListNode pCloned = null; //复制得到的链表的头结点
        RandomListNode pCNext = null;
        Map<RandomListNode,RandomListNode> mapNodes = new HashMap<RandomListNode,RandomListNode>();
        
        while(pNode != null) {
            if(pNode == pHead) {
                pCloned = new RandomListNode(pNode.label);//复制头结点
                pCNext = pCloned;
            } else {
                pCNext.next = new RandomListNode(pNode.label);//复制一个结点
                pCNext = pCNext.next;
            }
            mapNodes.put(pNode, pCNext);
            pNode = pNode.next;
            
        }
        
        //开始连接random
        Set<RandomListNode> s = mapNodes.keySet(); //取出键集合
        Iterator<RandomListNode> it = s.iterator();
        while(it.hasNext()) {
            RandomListNode rNode = it.next(); //取出原链表的结点
            if(rNode.random != null) {
                RandomListNode rCloned = mapNodes.get(rNode); //取出原链表结点的克隆结点
                rCloned.random = mapNodes.get(rNode.random); //连接random
            }
        }
        
        return pCloned;
    }
}