复杂链表的复制-牛客JZ25

原题链接

原题：

描述
输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针random指向一个随机节点），请对此链表进行深拷贝，并返回拷贝后的头结点。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）。 下图是一个含有5个结点的复杂链表。图中实线箭头表示next指针，虚线箭头表示random指针。为简单起见，指向null的指针没有画出。

示例:
输入:{1,2,3,4,5,3,5,#,2,#}
输出:{1,2,3,4,5,3,5,#,2,#}
解析:我们将链表分为两段，前半部分{1,2,3,4,5}为ListNode，后半部分{3,5,#,2,#}是随机指针域表示。
以上示例前半部分可以表示链表为的ListNode:1->2->3->4->5
后半部分，3，5，#，2，#分别的表示为
1的位置指向3，2的位置指向5，3的位置指向null，4的位置指向2，5的位置指向null
如下图:

示例1
输入：
{1,2,3,4,5,3,5,#,2,#}
返回值：
{1,2,3,4,5,3,5,#,2,#}

Code

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

    RandomListNode(int label) {
        this.label = label;
    }
}
*/
import java.util.*;

public class Solution {
    public RandomListNode Clone(RandomListNode pHead) {
        RandomListNode newHead = new RandomListNode(-1);
        // 存储原链表
        List<RandomListNode> list1 = new ArrayList();
        // 存储新链表
        List<RandomListNode> list2 = new ArrayList();
        // 存储原链表每个节点下标
        Map<RandomListNode , Integer> map = new HashMap();
        int j=0;
        while(pHead != null) {
            list1.add(pHead);
            list2.add(new RandomListNode(pHead.label));
            map.put(pHead , j++);
            pHead = pHead.next;
        }
        Map<Integer , Integer> randomMap = new HashMap();
        for(int i=0; i<list1.size(); i++) {
            RandomListNode cur = list1.get(i);
            if(cur.random != null) {
                // 存储cur指向cur.radom的位置
                randomMap.put(i , map.get(cur.random));
            }
        }
        RandomListNode p = newHead;
        for(int i=0; i<list2.size(); i++) {
            RandomListNode before = list1.get(i);
            RandomListNode now = list2.get(i);
            // 处理新链表中的节点random
            if(before.random != null) {
                now.random = list2.get(randomMap.get(i));
            }
            p.next = now;
            p = p.next;
        }
        return newHead.next;
    }
}