【leecode 剑指offer】复杂链表的深拷贝

复杂链表的深拷贝

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

解析

用一个哈希表表示映射关系：键是原节点，值是复制的节点。

整体算法流程是：

• 第一次遍历，复制每个节点和 next 指针，并且保存“原节点-复制节点”的映射关系
• 第二次遍历，通过哈希表获得节点对应的复制节点，更新 random 指针

代码实现

/**
 * // Definition for a Node.
 * function Node(val, next, random) {
 *    this.val = val;
 *    this.next = next;
 *    this.random = random;
 * };
 */

/**
 * @param {Node} head
 * @return {Node}
 */
var copyRandomList = function(head) {
    if(!head){
        return null;
    }
    let node = head;//记录当前节点
    let newHead = new Node(node.val);//创建新链表用于拷贝
    let newNode = newHead;//记录新列表的当前节点
    let map = new Map();//创建哈希表，用于记录新旧节点的对应关系
    map.set(node,newNode);//记录新旧节点的对应关系
    
    //第一次遍历，深度拷贝原链表的值与next指针
    while(node.next!=null){
        newNode.next=new Node(node.next.val);
        node = node.next;
        newNode = newNode.next;
        map.set(node,newNode);
    }
    //复原当前节点
    node=head;
    newNode=newHead;
    //第二次遍历，利用哈希表深度拷贝random指针
    while(node!=null){
        newNode.random = map.get(node.random);//获取新旧节点的对应关系
        node=node.next;
        newNode=newNode.next;
    }
    // 返回结果
    return newHead;
   
};