35-复杂链表的复制
2020-05-06 本文已影响0人
一方乌鸦
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
考虑用哈希表
class Solution {
public Node copyRandomList(Node head) {
if (head == null) return null;
Map<Node, Node> map = new HashMap<>();
Node p = head;
while(p != null) {
map.put(p, new Node(p.val));
p = p.next;
}
p = head;
while(p != null) {
Node n = map.get(p);
n.next = map.get(p.next);
n.random = map.get(p.random);
p = p.next;
}
return map.get(head);
}
}