leetcode 复制带随机指针的链表 题解

题目描述

给定一个链表，每个节点包含一个额外增加的随机指针，该指针可以指向链表中的任何节点或空节点。

要求返回这个链表的深拷贝。 

 

示例：

输入：
{"$id":"1","next":{"$id":"2","next":null,"random":{"$ref":"2"},"val":2},"random":{"$ref":"2"},"val":1}

解释：
节点 1 的值是 1，它的下一个指针和随机指针都指向节点 2 。
节点 2 的值是 2，它的下一个指针指向 null，随机指针指向它自己。
 

提示：

你必须返回给定头的拷贝作为对克隆列表的引用。

解题思路

使用hashmap储存原始链表节点和新建节点，两次遍历链表，第一次为新建节点赋值，第二次为新建节点指定next和random指定指向。

代码

/*
// Definition for a Node.
class Node {
    public int val;
    public Node next;
    public Node random;

    public Node() {}

    public Node(int _val,Node _next,Node _random) {
        val = _val;
        next = _next;
        random = _random;
    }
};
*/
class Solution {
    public Node copyRandomList(Node head) {
        if(head==null){
            return null;
        }; 
        HashMap<Node,Node> map =new HashMap<>();
        Node tmp = head;
        while(tmp!=null){
            map.put(tmp,new Node(tmp.val,null,null));
            tmp = tmp.next;
        }
        for(Node key: map.keySet()){
            tmp = map.get(key);
            tmp.next = map.get(key.next);
            tmp.random = map.get(key.random);
        }
        return map.get(head); 
    }
}