leetcode310--最小高度树

这道题需要用拓扑排序的思想做。我们可以把树想象成图，图想象成一层一层的圆，第一层是出度为1的节点，很容易知道，这些点并不是我们要找的最小高度节点（因为连接的那个点高度肯定小于等于他）。我们先把初始图初度为1的点加入队列，然后用BFS算法，依次把点和边删除，如果删除边之后出现出度为1的点再加入队列中，直到队列中的点与剩余未访问点的个数相同，这些点就是最后答案的点。

// 拓扑排序
class Solution {
public:
    vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {
        if (n == 1) return {0};
        // 根据边表edges 创建图(邻接数组)
        vector<vector<short>> adjList(n);
        vector<short> degree(n, 0);
        for (short i=0; i < edges.size(); ++i) {
            short v1 = edges[i][0];
            short v2 = edges[i][1];
            adjList[v1].push_back(v2);
            adjList[v2].push_back(v1);

            ++degree[v1];
            ++degree[v2];
        }

        // BFS拓扑排序
        // 初始化
        queue<short> Q;
        for (short i=0; i < n; ++i) {
            if (degree[i] == 1) {
                Q.push(i);
            }
        }
        
        while (Q.size()) {
            short size = Q.size();
            if (size == n) break; // 终止条件
            n -= size;

            while (size--) {
                short cur = Q.front(); Q.pop();

                // 访问邻接点，找出正确的下一个adj
                short adj;
                for (short i=0; i < adjList[cur].size(); ++i) {
                    adj = adjList[cur][i];
                    if (degree[adj] > 1) break; // 度=1的是已经被访问过的
                }
                --degree[adj];

                if (degree[adj] == 1) {
                    Q.push(adj);
                }
            }
        }

        vector<int> ans;
        while (Q.size()) {
            ans.push_back(Q.front());
            Q.pop();
        }
        
        return ans;
    }
};