leetcode最小高度树

使用了BFS/拓扑排序

class Solution {
public:
    vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges)
    {
        //只有一个根结点
        if (n == 1)
        {
            return {0};
        }
        //有两个结点，可以互为根结点
        if (n == 2)
        {
            return {0, 1};
        }
        //存放图
        vector<int> v;
        vector<vector<int>> graph(n, v);
        //存放每个结点度的数量
        vector<int> degree(n, 0);
        //初始化无向图
        for (int i = 0; i < edges.size(); i++)
        {
            graph[edges[i][0]].push_back(edges[i][1]);
            graph[edges[i][1]].push_back(edges[i][0]);
            degree[edges[i][0]]++;
            degree[edges[i][1]]++;
        }
        queue<int> q;
        for (int i = 0; i < n; i++)
        {
            //将度为1的结点全部入队
            if (degree[i] == 1)
            {
                q.push(i);
            }
        }
        int len = q.size();
        //大于两个结点时
        while(n > 2)
        {
            //一次性删除全部度为1的结点
            n -= len;
            while(len--)
            {
                //获取该节点
                int temp = q.front();
                //将该结点出队
                q.pop();
                //该结点度变为0
                degree[temp] = 0;
                for (int i = 0; i < graph[temp].size(); i++)
                {
                    //如果和temp结点有关联的结点没有被删除的话（即度为0）
                    if (degree[graph[temp][i]] != 0)
                    {
                        //那么因为temp被删除了，所以相关联的结点的度减1
                        degree[graph[temp][i]]--;
                        //度减一后变为1的话入队
                        if (degree[graph[temp][i]] == 1)
                        {
                            q.push(graph[temp][i]);
                        }
                    }
                }
            }
            len = q.size();
        }
        vector<int> result;
        while (!q.empty())
        {
            result.push_back(q.front());
            q.pop();
        }
    return result;
    }
};