Leetcode 310. Minimum Height Trees

Leetcode  310. Minimum Height Trees

这个解法思路很巧妙：

从最简单地开始：如果是一个list，从两端采用双指针的方法，当两个指针相遇或间隔为1时返回。

如果是一个树的话，则从度为1的节点（不一定是叶子节点），度为一只需要统计边数就可以了，多个指针向前遍历（宽度优先搜索），直到只剩一个或两个节点，便为所求。

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class Solution {
	public List<Integer> findMinHeightTrees(int n, int[][] edges) {
	    if(n==1)  return Collections.singletonList(0);
	    
	    List<Set<Integer>>  graph = new ArrayList<Set<Integer>>();
	    for(int i=0;i<n;i++)  graph.add(new HashSet<Integer>());
	    for(int[] edge:edges){
	    	graph.get(edge[0]).add(edge[1]);
	    	graph.get(edge[1]).add(edge[0]);
	    }
	    
	    List<Integer> leaves = new ArrayList<Integer>();
	    for(int i=0;i<n;i++){
	    	if(graph.get(i).size()==1)  leaves.add(i);
	    }
	    
	    while(n>2){
	    	n-=leaves.size();
	    	List<Integer> newleaves = new ArrayList<Integer>();
	    	for(int i :leaves){
	    		int j = graph.get(i).iterator().next();
	    		graph.get(j).remove(i);
	    		if(graph.get(j).size()==1)  newleaves.add(j);
	    	}
	    	leaves = newleaves;
	    }
	    return leaves;
	}
}