题目大意:给出一颗无根树,问以哪个节点为根能使得树的高度最小
分析:拓扑排序。要找最小高度的根节点,就是找整个图中最中心的点,也就是距离所有叶节点最远的结点。拓扑排序的思想,每次都取出图中入度为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;
}
};