對于一個具有樹特征的無向圖，我們可選擇任何一個節點作為根。圖因此可以成為樹，在所有可能的樹中，具有最小高度的樹被稱為最小高度樹。給出這樣的一個圖，寫出一個函數找到所有的最小高度樹并返回他們的根節點。格式該圖包含 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]

代碼

class Solution {public List<Integer> findMinHeightTrees(int n, int[][] edges) {Map<Integer,List<Integer>> map=new HashMap<>();for(int i=0;i<n;i++)map.put(i,new ArrayList<>());for (int[] edge:edges)//建立鄰接表{map.get(edge[0]).add(edge[1]);map.get(edge[1]).add(edge[0]);}boolean[] check=new boolean[n]; int[] degree=new int[n];Queue<Integer> queue=new LinkedList<>();for(int i=0;i<n;i++)//記錄節點的度，并且將度為1的節點入隊if (map.get(i).size()==1) {queue.add(i);check[i]=true;degree[i]=1;}else {degree[i]=map.get(i).size();}if(queue.isEmpty()) return new ArrayList<>(){{add(0);}};//沒有度為1的節點ArrayList<Integer> res=new ArrayList<>();while (!queue.isEmpty())//bfs{int size=queue.size();ArrayList<Integer> temp=new ArrayList<>();for(int i=0;i<size;i++){int cur=queue.poll();for(int c:map.get(cur))//移除當前節點，將相鄰節點的度減去1{if(!check[c]){degree[c]--;if(degree[c]<=1)//度為1則入隊{queue.offer(c);check[c]=true;}}}temp.add(cur);}res=temp;}return res;} }

總結

以上是生活随笔為你收集整理的leetcode310. 最小高度树（bfs）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。