无向图的最小高度树

原题在这里：

　　概述题意：给定节点数和无向图边，问能构成最小高度树的所有根节点。

analyse:

　　1.最直白的方式就是bfs遍历找当前节点的最长的最短路。ans选为该(最长的最短路)最短节点集合。　　

　　　　但是数据范围不适合，考虑优化，我想当然式的以为从边数最多节点开始遍历直到深度变大(以为从边数最多开始一定最小，这是错误的)

　　2.标准容易的解法，确实是根据边数，每次循环边数为1删边，直到最后一组边即为根节点

　　　　画个树图就懂了，也就是不断删去树的叶子节点，最后肯定是根节点了。

code：

 

class Solution
{
public:
    vector<int> findMinHeightTrees(int n, vector<vector<int>> &edges)
    {
        if (n == 1)
            return {0};
        vector<int> num(n), ans;
        vector<vector<int>> mp(n);
        queue<int> q;
        for (auto i : edges)
        {
            ++num[i[0]];
            ++num[i[1]];
            mp[i[0]].emplace_back(i[1]);
            mp[i[1]].emplace_back(i[0]);
        }
        for (int i = 0; i < n; i++)
            if (num[i] == 1)
                q.push(i);
        while (!q.empty())
        {
            ans.clear();
            int m = q.size();
            while (m--)
            {
                int x = q.front();
                q.pop();
                ans.push_back(x);
                num[x]--;
                for (auto y : mp[x])
                    if (--num[y] == 1)
                        q.push(y);
            }
        }
        return ans;
    }
};

 

 

 

【Over】