310. 最小高度树

树是一个无向图，其中任何两个顶点只通过一条路径连接。 换句话说，一个任何没有简单环路的连通图都是一棵树。

给你一棵包含 n 个节点的树，标记为 0 到 n - 1 。给定数字 n 和一个有 n - 1 条无向边的 edges 列表（每一个边都是一对标签），其中 edges[i] = [ai, bi] 表示树中节点 ai 和 bi 之间存在一条无向边。

可选择树中任何一个节点作为根。当选择节点 x 作为根节点时，设结果树的高度为 h 。在所有可能的树中，具有最小高度的树（即，min(h)）被称为 最小高度树 。

请你找到所有的 最小高度树 并按 任意顺序 返回它们的根节点标签列表。

树的 高度 是指根节点和叶子节点之间最长向下路径上边的数量。

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/minimum-height-trees

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

import java.util.*;

class Solution {
    public List<Integer> findMinHeightTrees(int n, int[][] edges) {
        if (n == 1) {
            return Collections.singletonList(0);
        }
        List<Integer> ans = new ArrayList<>();
        int[] in = new int[n];
        Map<Integer, Set<Integer>> graph = new HashMap<>();
        for (int[] edge : edges) {
            in[edge[0]]++;
            in[edge[1]]++;
            graph.computeIfAbsent(edge[0], k -> new HashSet<>()).add(edge[1]);
            graph.computeIfAbsent(edge[1], k -> new HashSet<>()).add(edge[0]);
        }
        Queue<Integer> queue = new LinkedList<>();
        for (int i = 0; i < in.length; ++i) {
            if (in[i] == 1) {
                queue.offer(i);
            }
        }

        while (!queue.isEmpty()) {
            ans = new ArrayList<>();
            int size = queue.size();
            while (size-- > 0) {
                Integer node = queue.poll();
                ans.add(node);
                Set<Integer> tos = graph.getOrDefault(node, Collections.emptySet());
                for (int to : tos) {
                    if (--in[to] == 1) {
                        queue.offer(to);
                    }
                }
            }
        }

        return ans;
    }
}
           

心之所向，素履以往 生如逆旅，一苇以航