133. 克隆图

克隆图

给定无向连通图中一个节点的引用,返回该图的深拷贝(克隆)。图中的每个节点都包含它的值 val(Int) 和其邻居的列表(list[Node])。

示例:
133. 克隆图_第1张图片

输入:
{"$id":"1","neighbors":[{"$id":"2","neighbors":[{"$ref":"1"},{"$id":"3","neighbors":[{"$ref":"2"},{"$id":"4","neighbors":[{"$ref":"3"},{"$ref":"1"}],"val":4}],"val":3}],"val":2},{"$ref":"4"}],"val":1}

解释:
节点 1 的值是 1,它有两个邻居:节点 2 和 4 。
节点 2 的值是 2,它有两个邻居:节点 1 和 3 。
节点 3 的值是 3,它有两个邻居:节点 2 和 4 。
节点 4 的值是 4,它有两个邻居:节点 1 和 3 。

思路+代码+注释:

public Node cloneGraph(Node node) {
           /*
           思路:使用hashMap存储已经拷贝过的节点,其中key为原节点,value为拷贝节点
           递归扫描节点,如果该节点在map中不存在说明没有被拷贝过,那么新建一个节点添加到map中,在递归克隆邻居节点;如果该节点在map中,说明被拷贝过直接返回
           当节点为null时也直接返回null
            */
        HashMap nodeMap=new HashMap<>();
        return cloneGraph(node,nodeMap);
    }

    private Node cloneGraph(Node node, HashMap map)
    {
        if (node==null)
        {
            return null;
        }
        if (map.containsKey(node))
        {
            return map.get(node);
        }
        Node newNode=new Node();
        newNode.val=node.val;
        map.put(node,newNode);
        List newList=new ArrayList<>();
        List nodeList=node.neighbors;
        for (int i = 0; i < nodeList.size(); i++) {
            Node oldNode=nodeList.get(i);
            newList.add(cloneGraph(oldNode,map));
        }
        newNode.neighbors=newList;
        return newNode;
    }

你可能感兴趣的:(LeetCode)