题目
请实现 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]]
思路
一般思路是先复制节点的val值与next指针,然后再复制random指针(为什么不同时复制呢?因为可能当前节点的random指向的是比较后面的节点,但是此时并没有这个节点)
所以此时时间复杂度为O(n^2)
可以使用hash表进行复制:利用哈希表的查询特点,考虑构建 原链表节点 和 新链表对应节点 的键值对映射关系,再遍历同时构建新链表各节点的 next 和 random 引用指向即可。可看力扣上面的详解
代码
var copyRandomList = function(head) {
if(!head) return null
let p=head
let nodemap=new Map()
let t
while(p) {
t = new Node(p.val,null,null)
nodemap.set(p,t)
p = p.next
}
p = head;
nodemap.set(null,null)//这句话很重要,因为在ES6的Map中,不存在键值对(NULL,NULL),在下面的get(p.next)和get(p.random)中可能取到NULL,因此需要加上这一句。
while (p) {
nodemap.get(p).next = nodemap.get(p.next)
nodemap.get(p).random = nodemap.get(p.random)
p = p.next
}
return nodemap.get(head)
};