【Leetcode】310. Minimum Height Trees(转载,以便查询)

167 阅读2分钟

题目地址: leetcode.com/problems/mi…

给定一个n nn阶简单无向图,其有树性质。若取其中某个顶点为根,其就成为一棵有根树,从而有树高。求使得树高最小的顶点编号。答案也许不唯一。

思路是BFS。如果图平凡(意思是图只有一个顶点)则直接返回0。如果图的顶点数大于1,直观的想法是,可以先将度为1的点入队,然后向树的“内部”分层遍历,然后将这些点删掉,再将剩余的点中度为1的点入队遍历,直到整个图被遍历完,这样最后一层遍历到的顶点即为所求(整个过程有点像拓扑排序,只不过这里的图是无向图,所以要不停地将度为1的点入队)。

算法正确性证明: 数学归纳法。对图的顶点数目进行归纳。如果图的顶点数为2,那么两个顶点的度都是1,那么这两个顶点既是最外层的顶点,也是最后一层被遍历的顶点,而算法返回的正是这两个顶点,所以算法正确(如果图顶点数为1 的话类似)。 对于图的顶点数V > 2 的情况,我们知道图的所有顶点的度的总和等于2 E = 2 ( V − 1 ) = 2 V − 2 ,首先,度为1 的顶点一定是存在的,否则度数和就会大于等于2 V ,矛盾;其次,由于2 V − 2 > V ,所以图中存在度不是1的顶点,也就是说,删掉度为1的顶点之后图仍然非空(也就是仍然有顶点);以上结论是为做数学归纳法做准备。 接着我们只需要证明“最外层”的顶点不可能成为最矮树的树根就行。任取其中一个顶点v ,设离v 最远的顶点是u ,那么树的高就是两者的距离,设为h 。设v 的邻接点是w (按照度的定义,w 是v 的唯一的邻接点),若以w 为树根,则高度为h − 1 ,更小。所以v 不可能是最矮树的树根。把“最外层”的顶点删除后,图的顶点数减少,但仍然不空(上面的结论),由数学归纳法知道,算法对其正确。综上,算法正确。

由这个算法也能知道,最终的答案数目最多是2 。不然的话,由上面的证明,仍然存在度为1 的最外层的顶点还没删掉,这就矛盾了。所以最终的答案数目不大于2 。

import java.util.*;

public class Solution {
    public List<Integer> findMinHeightTrees(int n, int[][] edges) {
        List<Integer> res = new ArrayList<>();
        
        // 特判是平凡图的情形
        if (n == 1) {
            res.add(0);
            return res;
        }
        
        // 邻接表建图,并算出每个顶点的度
        Map<Integer, List<Integer>> graph = buildGraph(edges);
        int[] degrees = getDegrees(n, edges);
        
        // 将度为1的点全部入队
        Queue<Integer> queue = new LinkedList<>();
        for (int i = 0; i < degrees.length; i++) {
            if (degrees[i] == 1) {
                queue.offer(i);
            }
        }
        
        while (!queue.isEmpty()) {
        	// 先把这一层的顶点全加入答案中
            res.clear();
            res.addAll(queue);
            
            // 开始分层遍历,由于要分层,所以要记录一下队列的大小
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                int cur = queue.poll();

                // 得到cur的邻接点
                for (int next : graph.get(cur)) {   		
              		// 将next的度减少1,然后将度为1的顶点入队      
                    degrees[next]--;
                    if (degrees[next] == 1) {
                        queue.offer(next);
                    }
                }
            }
        }
        
        // 队列为空的时候,最后访问的那一层顶点即为所求
        return res;
    }
    
    private int[] getDegrees(int n, int[][] edges) {
        int[] degrees = new int[n];
        for (int[] edge : edges) {
            int x = edge[0], y = edge[1];
            degrees[x]++;
            degrees[y]++;
        }
        
        return degrees;
    }
    
    private Map<Integer, List<Integer>> buildGraph(int[][] edges) {
        Map<Integer, List<Integer>> graph = new HashMap<>();
        for (int[] edge : edges) {
            int x = edge[0], y = edge[1];
            graph.computeIfAbsent(x, k -> new ArrayList<>()).add(y);
            graph.computeIfAbsent(y, k -> new ArrayList<>()).add(x);
        }
        
        return graph;
    }
}

原文地址:blog.csdn.net/qq_46105170…