【leetcode】310. Minimum Height Trees

解决思路

以这棵树为例，首先MHT的根节点不会是叶子节点，所以删除0，1，2，5
剩下的3，4就是MHT的根节点。

即一层一层的删掉叶子节点，直到只剩下一个或者两个节点。

提交代码

class Solution{
	 public List<Integer> findMinHeightTrees(int n, int[][] edges) {
		 if(n==1) return Collections.singletonList(0);
		 
		 List<Set<Integer>> tree = new ArrayList<>(n);
		 for(int i =0;i<n;i++)
			 tree.add(new HashSet<>());
		 
		 for(int[] edge:edges) {
			 tree.get(edge[0]).add(edge[1]);
			 tree.get(edge[1]).add(edge[0]);
		 }
		 
		 int treeSize = n;
		 List<Integer> leaves = new ArrayList<>();

		 while(treeSize>2) {
			 leaves.clear();
			 for(int i=0;i<tree.size();i++) 
				 if(tree.get(i).size()==1) 
					 leaves.add(i);
			 
			 treeSize -= leaves.size();
			 if(treeSize < 3) {
				 for(int i : leaves) 
					 tree.get(i).clear();
				 break;
			 }
			 
			 for(int i : leaves) {
				 tree.get(tree.get(i).iterator().next()).remove(i);
				 tree.get(i).clear();
			 }
		 }
		 
		 List<Integer> res = new ArrayList<>();
		 for(int i=0;i<n;i++) {
			 if(tree.get(i).size()!=0)
				 res.add(i);
		 }
		 return res;
 	 }
}

运行结果