leetcode -- Minimum Height Trees -- 关于graph的，重点

xyqzki

于 2015-12-10 22:29:42 发布

阅读量2.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/xyqzki/article/details/50255923

leetcode 专栏收录该内容

257 篇文章

订阅专栏

本文介绍了一种寻找图中最小高度树的有效算法。通过逐步移除叶子节点的方式找到树的根节点，适用于解决LeetCode上的Minimum Height Trees问题。文章详细解释了算法原理，并提供了Python实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://leetcode.com/problems/minimum-height-trees/

有trick
对graph不熟悉。再看几遍。

参考http://bookshadow.com/weblog/2015/11/26/leetcode-minimum-height-trees/

基本思路是“逐层删去叶子节点，直到剩下根节点为止”

有点类似于拓扑排序

最终剩下的节点个数可能为1或者2

时间复杂度：O(n)，其中n为顶点的个数。

记住这些结论以及这样的code怎么写就行了。就跟拓扑排序一样，逐次去掉leaf node

class Solution(object):
    def findMinHeightTrees(self, n, edges):
        """
        :type n: int
        :type edges: List[List[int]]
        :rtype: List[int]
        """
        children = collections.defaultdict(set)
        for s, t in edges:#记录edge
            children[s].add(t)
            children[t].add(s)
        vertices = set(children.keys())
        while len(vertices) > 2:
            leaves = [x for x in children if len(children[x]) == 1]
            for x in leaves:
                for y in children[x]:
                    children[y].remove(x)
                del children[x]
                vertices.remove(x)
        return list(vertices) if n != 1 else [0]