leetcode310——Minimum Height Trees

题目大意：给出一颗无根树，问以哪个节点为根能使得树的高度最小

分析：拓扑排序。要找最小高度的根节点，就是找整个图中最中心的点，也就是距离所有叶节点最远的结点。拓扑排序的思想，每次都取出图中入度为1的结点，删除这些节点并将与他们相邻的结点和他们之间的边删掉，就像剥洋葱一样，逐层删除叶节点，这样最后剩下的结点就是最中心的结点（只可能是一个或者两个）

代码：转载自http://lib.csdn.net/article/datastructure/17377

class Solution {
public:
    vector<int> findMinHeightTrees(int n, vector<pair<int, int>>& edges) {
        vector<int> indegree(n,0), ans;
        for(auto val: edges) indegree[val.first]++, indegree[val.second]++;
        unordered_map<int, set<int>> hash;
        for(auto val: edges)
        {
            hash[val.first].insert(val.second);
            hash[val.second].insert(val.first);
        }
        while(n > 2)
        {
            vector<int> vec;
            for(int i =0; i < indegree.size(); i++) 
                if(indegree[i]==1) vec.push_back(i);
            for(auto k: vec)
            {
                indegree[k] = -1, n--;
                for(auto val: hash[k]) indegree[val]--;
            }
        }
        for(int i = 0; i < indegree.size(); i++)
            if(indegree[i] >= 0) ans.push_back(i);
        return ans;
    }
};