LeetCode_310最小高度树（拓扑排序）

题目链接：https://leetcode-cn.com/problems/minimum-height-trees/description/

 

题目大意：

        把一个无向图想象成一棵树，每个结点都可以是树的根节点，然后找到最小的树的高度所对应的根结点。

思路解法：

        开始把这个题想得有点简单，我用dfs计算出了每个点作为根节点时的树的高度，然后用当前最小高度来剪枝。然而超时了，后来想想确实dfs的时间复杂度太高。后来在网上看到大神的思路，利用了拓扑排序的思想，简单易懂，复杂度低。拓扑排序的详解：https://blog.csdn.net/Q_M_X_D_D_/article/details/84862081

        本题的解法思路是：设立一个点集，保存当前图中度为1的点，即树的叶子结点。然后将这些结点从图中删去，此时，有可能会生成一些新的叶子结点，那么再将这些新的叶子结点加入点集中。不断重复这个过程，直到图中的剩下的点不超过3个。为什么是3个呢？举个例子，假设一个图有两个点，用一条边连起来，那么返回的结果就是这两个点。但如果图中有三个点，用两条边连起来，那么返回的结果就是中间的那一个点。代码如下：

class Solution {
public:
    vector<int> findMinHeightTrees(int n, vector<pair<int, int>>& edges) {
    	vector<int> res;
    	//储存每个点的度 
    	vector<int> indegree(n,0);
    	//储存图中边的信息 
        unordered_map<int,vector<int> > hash;
        for(int i=0;i<edges.size();i++)
        {
        	//建图并统计点的度 
        	hash[edges[i].first].push_back(edges[i].second);
        	hash[edges[i].second].push_back(edges[i].first);
        	indegree[edges[i].first]++;
        	indegree[edges[i].second]++;
		}
		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++)
			{
				//删去这个叶子结点，并将与之相连的点的度-1 
				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)
				res.push_back(i);
		return res;
    }
};