初学python记录：力扣310. 最小高度树

题目：

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

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

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

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

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

思考：

最开始想的是暴力枚举，但是这样时间复杂度就到O(n²)了，肯定会超时。

参考了题解的最后一种解法和知乎大佬的讲解，选择用类似拓扑排序的方法：

找出最小高度树的根节点 ----> 找出距离其他所有节点最近的节点

将图类比为一个圆，则要找到圆的质心。

对于树，最多有两个“质心”（因为树中没有环）。

要找到“质心”，则需要从外围向内部：修剪叶子节点  ----> 产生新的叶子节点 ----> 修剪叶子节点 ----> .......，直至只剩下最后的一个或两个“质心”。

代码如下，需要注意区分python中对列表和集合的操作函数有所不同：

class Solution(object):
    def findMinHeightTrees(self, n, edges):
        """
        :type n: int
        :type edges: List[List[int]]
        :rtype: List[int]
        """
        # 若节点数小于等于2，则全为最小高度树的根节点
        if n <= 2:
            return [i for i in range(0,n)]
        # 类似拓扑排序---->层层剥离
        # 1.初始化邻接表
        graphic = [set() for _ in range(0, n)]

        # 根据每一条边，填充邻接表
        for x,y in edges:
            graphic[x].add(y)
            graphic[y].add(x)

        # 初始化叶子节点列表，把所有只有一个相邻节点的节点加入列表
        leaf = []
        for i in range(0, n):
            if len(graphic[i]) == 1:
                leaf.append(i)
        
        while(n>2):
            new_leaf = []
            n -= len(leaf)
            # 逐个剥离叶子节点
            for node in leaf:
                # 将叶子节点的邻接节点集合置空，并得到与其唯一相邻的节点
                neighbor = graphic[node].pop()
                # 在叶子节点相邻的节点的邻接节点集合中将其删除
                graphic[neighbor].remove(node)   
                # 若使这个相邻节点变成叶子节点，则加入新的叶子节点列表
                if len(graphic[neighbor]) == 1:
                    new_leaf.append(neighbor)
            leaf = new_leaf
        
        return [i for i in leaf]
            

运行通过：

参考：

 LeetCode 310. 最小高度树 (Minimum Height Trees) - 知乎 (zhihu.com)