最小高度的樹 Minimum Height Trees

標籤：bfs   lis   使用   好的   return   leave   div   圖片   list   

2018-09-24 12:01:38

問題描述：

問題求解：

毫無疑問的一條非常好的題目，採用的解法是逆向的BFS，也就是從葉子節點開始遍曆，逐步向中心靠攏，最終留下的葉子節點就是答案。

    public List<Integer> findMinHeightTrees(int n, int[][] edges) {        // tricks ：使用Collections.singletonList(0)可以直接返回單個值的List        if (n == 1) return Collections.singletonList(0);        // tricks ：使用Set是為了方便在之後維護graph的時候方便按值刪除        List<Set<Integer>> graph = new ArrayList<>(n);        for (int i = 0; i < n; i++) graph.add(new HashSet<>());        for (int[] edge : edges) {            graph.get(edge[0]).add(edge[1]);            graph.get(edge[1]).add(edge[0]);        }        List<Integer> leaves = new ArrayList<>();        for (int i = 0; i < n; i++) {            if (graph.get(i).size() == 1) leaves.add(i);        }        while (n > 2) {            n -= leaves.size();            List<Integer> newLeaves = new ArrayList<>();            for (int i : leaves) {                int j = graph.get(i).iterator().next();                graph.get(j).remove(i);                if (graph.get(j).size() == 1) newLeaves.add(j);            }            leaves = newLeaves;        }        return leaves;    }

 

最小高度的樹 Minimum Height Trees