C/C++教程

138. Copy List with Random Pointer

本文主要是介绍138. Copy List with Random Pointer,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

这道题咋看挺复杂,又要clone node还要clone random link,但其实只要用一个HashMap就可以轻松解决,以下是我的算法,先clone node和node的next link,然后clone node的random link,时间复杂度O(n):

    public Node copyRandomList(Node head) {
        if (head == null)
            return null;
        Map<Node, Node> map = new HashMap<>();
        Node res = new Node(head.val);
        clone(head, res, map);
        randomLink(head, res, map);
        return res;
    }

    private void clone(Node oldNode, Node newNode, Map<Node, Node> map) {
        map.put(oldNode, newNode);
        if (oldNode.next != null) {
            newNode.next = new Node(oldNode.next.val);
            clone(oldNode.next, newNode.next, map);
        }
    }

    private void randomLink(Node oldNode, Node newNode, Map<Node, Node> map) {
        if (oldNode.random != null) {
            newNode.random = map.get(oldNode.random);
        }
        if (oldNode.next != null) {
            randomLink(oldNode.next, newNode.next, map);
        }
    }

 

这篇关于138. Copy List with Random Pointer的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!