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);
    }
}
上一篇下一篇

猜你喜欢

热点阅读