leetcode 310. Minimum Height Trees 解题报告

原题链接

原题链接

解题思路

常规方法可以使用BFS或者DFS，对每个点都遍历一遍，求出所有点组成的树的高度，然后找出哪些高度最小的节点，可以通过不断更新最低高度来进行剪枝。但是时间复杂度过高。
最终的解题思路采用了不断删除叶子节点，逐渐逼近根节点的方法，在删除叶子节点的同时，会有一些新的节点成为叶子节点，于是继续循环删除，直至不能删除为止，那么剩下的节点就是高度最小的根。
首先将图存储起来，并将每个节点的入度存储起来，然后遍历将入度为1（叶节点）的节点加入队列。
bfs，当队列不空，取队列存储过的叶子节点再将叶子节点的邻接节点，如果邻接节点的度也为1（叶子节点），加入队列。
如此反复，直到减掉所有叶子节点最后一批剩下来的就是正确答案。

解题代码

public class Solution {
    public List<Integer> findMinHeightTrees(int n, int[][] edges) {
        List<List<Integer>> map = new ArrayList<List<Integer>>();
        List<Integer> res = new ArrayList<Integer>();
        if (n==1) {
            res.add(0);
            return res;
        }
        //记录每个点的入度
        int[] degree = new int[n];
        for (int i = 0;i < n;i++ ) {
            map.add(new ArrayList<Integer>());
        }
        for (int i = 0;i< edges.length;i++) {
            map.get(edges[i][0]).add(edges[i][1]);
            map.get(edges[i][1]).add(edges[i][0]);
            degree[edges[i][0]]++;
            degree[edges[i][1]]++;
        }
        Queue<Integer> q = new ArrayDeque<Integer>();
        for (int i = 0;i < n;i++ ) {
            if(degree[i] == 0) {
                return res;
            } else if(degree[i] == 1) {
                q.offer(i);
            }
        }
        while(!q.isEmpty()) {
            res = new ArrayList<Integer>();
            int count = q.size();
            for (int i = 0;i < count ;i++ ) {
                int cur = q.poll();
                res.add(cur);
                degree[cur]--;
                for (int k = 0;k < map.get(cur).size() ;k++ ) {
                    int next = map.get(cur).get(k);
                    degree[next]--;
                    if(degree[next] == 0) continue;
                    if(degree[next] == 1) {
                        q.offer(next);
                    }
                }                
            }
        }
        return res;
    }
}