leetcode练习-310. 最小高度树(图+bfs+拓扑排序+贪心思想)

对于一个具有树特征的无向图，我们可选择任何一个节点作为根。图因此可以成为树，在所有可能的树中，具有最小高度的树被称为最小高度树。给出这样的一个图，写出一个函数找到所有的最小高度树并返回他们的根节点。

格式

该图包含 n 个节点，标记为 0 到 n - 1。给定数字 n 和一个无向边 edges 列表（每一个边都是一对标签）。

你可以假设没有重复的边会出现在 edges 中。由于所有的边都是无向边， [0, 1]和 [1, 0] 是相同的，因此不会同时出现在 edges 里。

示例 1:

输入: n = 4, edges = [[1, 0], [1, 2], [1, 3]]

    0
    |
    1
   / \
  2   3 

输出: [1]
示例 2:

输入: n = 6, edges = [[0, 3], [1, 3], [2, 3], [4, 3], [5, 4]]

 0  1  2
  \ | /
    3
    |
    4
    |
    5 

输出: [3, 4]
说明:

根据树的定义，树是一个无向图，其中任何两个顶点只通过一条路径连接。 换句话说，一个任何没有简单环路的连通图都是一棵树。
树的高度是指根节点和叶子节点之间最长向下路径上边的数量。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-height-trees
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
叶子节点的标志就是入度度为1，我们每次所有减去以入度为1为一端的边，直到最后留下的两个或者一个就是我们要的答案。

import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import java.util.Set;
class Solution {
    public List<Integer> findMinHeightTrees(int n, int[][] edges) {
        List<Integer> ans=new ArrayList<>();
        //最终会有一个或者两个根节点
        if(n==0)
        	return ans;
        if(n<=2) {
        	for(int i=0;i<n;i++) {
        		ans.add(i);
        	}
        	return ans;
        }
        //建立入度的数组
        int indegree[]=new int[n];
        //每个结点相连的结点
        Set<Integer>[] set=new Set[n];
        //初始化
        for(int i=0;i<n;i++) {
        	set[i]=new HashSet<>();
        }
        //链接边并且输入每个结点的度
        for(int[] array:edges) {
        	int left=array[0];
        	int right=array[1];
        	//边的链接
        	set[left].add(right);
        	set[right].add(left);
        	//入度的存放
        	indegree[left]++;
        	indegree[right]++;
        }
        //开始找根节点
        Queue<Integer> queue=new LinkedList<>();
        //保存被访问过的结点
        boolean visited[]=new boolean[n];
        //入度为1的入队
        for(int i=0;i<n;i++) {
        	if(indegree[i]==1)
        		queue.offer(i);
        }
        //n<=2就是该出来了，最后两个或一个度为1的结点在队列里面
        while(n>2) {
        	int size=queue.size();
        	n-=size;
        	for(int i=0;i<size;i++) {
        		int t=queue.poll();
        		//置为已访问
        		visited[t]=true;
        		//减边
        		indegree[t]--;
        		//与之有关的全部减1
        		Set<Integer> s=set[t];
        		for(int a:s) {
        			if(!visited[a]) {
        				indegree[a]--;
        				if(indegree[a]==1)
        					queue.offer(a);
        			}
        		}
        	}
        }
        while(!queue.isEmpty()) {
        	ans.add(queue.poll());
        }
        return ans;
    }
}