题目:
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
示例1:
输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]示例2:
输入:head = [[1,1],[2,1]] 输出:[[1,1],[2,1]]示例3:
输入:head = [[3,null],[3,0],[3,null]] 输出:[[3,null],[3,0],[3,null]]示例4:
输入:head = [] 输出:[] 解释:给定的链表为空(空指针),因此返回 null。
(迭代 + 哈希 的方法实现)
代码:
var copyRandomList = function(head) {
if(head === null){
return head;
}
let cur = head;
let newHead = new Node();
let newCur = newHead;
let map = new Map();
while(cur){
newCur.val = cur.val;
newCur.next = cur.next ? new Node() : null;
map.set(cur, newCur);
cur = cur.next;
newCur = newCur.next;
}
newCur = newHead;
while(head){
newCur.random = head.random ? map.get(head.random) : null;
head = head.next;
newCur = newCur.next;
}
return newHead;
};
1.单独定义一个新链表的头结点newHead,方便后续能够轻松的返回头结点。
2.定义新链表,初始指向newHead。
3.定义一个哈希图map。
4.第一个while循环,先不看random,将原链表其他给复制过来,如值val、next指向。
newCur.next = cur.next ? new Node() : null;
// 因为要复制原链表, 所以新链表的每一个结点都必须是新的结点,不能是原链表上的结点
5.利用map.set ,将原链表的每个结点与新链表中每个复制这个对应结点的新结点 形成一个映射关系, 为了方便后续给新链表建立random的指向关系
6.在执行完第一个while后,重新让新链表newCur指向头结点。
7.第二个while是为了给新链表建立原链表中的random关系
newCur.random = head.random ? map.get(head.random) : null;
// 判断结点是否有random指向
// 如果有, 则利用map.get,找出原链表中该结点的random指向的结点在新链表中的对应映射的结点,让其与这个匹配到的结点形成random的指向关系
8.最终原链表中每个结点的val值、next指向关系、random指向关系都成功的复刻到了新链表newCur上,最后返回新链表的头结点newHead即可