题目大意:给出一颗无根树,问以哪个节点为根能使得树的高度最小
分析:拓扑排序。要找最小高度的根节点,就是找整个图中最中心的点,也就是距离所有叶节点最远的结点。拓扑排序的思想,每次都取出图中入度为1的结点,删除这些节点并将与他们相邻的结点和他们之间的边删掉,就像剥洋葱一样,逐层删除叶节点,这样最后剩下的结点就是最中心的结点(只可能是一个或者两个)
代码:
class Solution {
public:
vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {
vector<int> indegree(n,0), ans;
vector<vector<int> > hash(n);
for(int i = 0;i < edges.size();i++){
indegree[edges[i][0]]++;
indegree[edges[i][1]]++;
hash[edges[i][0]].push_back(edges[i][1]);
hash[edges[i][1]].push_back(edges[i][0]);
}
while(n > 2){
vector<int> vec;
for(int i = 0; i < indegree.size(); i++)
if(indegree[i] == 1) vec.push_back(i);
for(int i = 0;i < vec.size();i++){
indegree[vec[i]] = -1;
n--;
for(int j = 0;j < hash[vec[i]].size();j++)
indegree[hash[vec[i]][j]]--;
}
}
for(int i = 0; i < indegree.size(); i++)
if(indegree[i] >= 0) ans.push_back(i);
return ans;
}
};