310. 最小高度树

对于一个具有树特征的无向图，我们可选择任何一个节点作为根。图因此可以成为树，在所有可能的树中，具有最小高度的树被称为最小高度树。给出这样的一个图，写出一个函数找到所有的最小高度树并返回他们的根节点。

思路：

最小高度的树的根一定不会在叶子节点取得！(怎么判断是叶子呢？它的邻边数量为1)

每一轮修剪一次剩余的所有叶子节点，一旦节点数量只有1或者2，停止修剪，返回答案。

class Solution(object):
    def findMinHeightTrees(self, n, edges):
        """
        :type n: int
        :type edges: List[List[int]]
        :rtype: List[int]
        """
        graph = [set() for x in range(n)]   # 将每个结点的邻居单独放在一个集合里
        for e in edges:
            graph[e[0]].add(e[1])
            graph[e[1]].add(e[0])
        points = set([x for x in range(n)])
        while len(points) > 2:
            removed = []
            for u in points:           # 先遍历所有结点，找出只有一个邻居的结点保存起来。
                if len(graph[u]) == 1:
                    removed.append(u)
            for p in removed:         # 清楚此叶结点的踪迹！即找到p对应的结点q，并从q的邻居中删除p，points中删除p
                q = graph[p].pop()   
                graph[q].remove(p)
                points.remove(p)
        return list(points)

参考：
https://blog.youkuaiyun.com/weixin_37373020/article/details/81109439