最小高度的樹 Minimum Height Trees
阿新 • • 發佈:2018-09-24
答案 i++ 技術 add clas int ++ rem ext
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