剑指 Offer 35. 复杂链表的复制 JavaScript 实现

问题描述

请实现 copyRandomList 函数，复制一个复杂链表。在复杂链表中，每个节点除了有一个 next 指针指向下一个节点，还有一个 random 指针指向链表中的任意节点或者 null。

输入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出：[[7,null],[13,0],[11,4],[10,2],[1,0]]

代码

var copyRandomList = function(head) {
    if(head == null) return head;
    let map = new Map();
    cur = head;
    while(cur != null) { // 第一次现将父节点复制到子节点
        map.set(cur, new Node(cur.val));
        cur = cur.next;
    }
    cur = head;
    while(cur != null){ // 第二次复制将父节点之间的关系，复制到子节点上
        let newNode = map.get(cur);
        let next = map.get(cur.next);
        let random = map.get(cur.random);
        if (next) newNode.next = next;
        if (random) newNode.random = random;
        cur = cur.next;
    }
    return map.get(head);// 最后返回子链表的头结点
};