对于一个具有树特征的无向图,我们可选择任何一个节点作为根。图因此可以成为树,在所有可能的树中,具有最小高度的树被称为最小高度树。给出这样的一个图,写出一个函数找到所有的最小高度树并返回他们的根节点。
格式
该图包含 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;
}
}