问题描述
请实现 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);// 最后返回子链表的头结点
};