力扣(LeetCode)310

題目地址：
https://leetcode-cn.com/probl...
題目描述：

對於一個具備樹特徵的無向圖，咱們可選擇任何一個節點做爲根。圖所以能夠成爲樹，在全部可能的樹中，具備最小高度的樹被稱爲最小高度樹。給出這樣的一個圖，寫出一個函數找到全部的最小高度樹並返回他們的根節點。

格式

該圖包含 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]

解答：
這一題比較有技巧，若是求任意一點到任意一點的距離，那麼會時間複雜度會很大。
比較高效的作法是，每次把葉子節點從圖(把樹轉換爲圖結構)刪掉。
直到只剩下1個或者2個點的時候輸出。

算法思想很簡單。可是實現起來有寫麻煩。若是每次都判斷葉子節點，那麼效率會很低。
所以使用一個數組inDegree[]表明每一個節點的入度，若入度爲1就是葉子節點。
而且用一個隊列(棧也能夠，只要是那種能彈出的容器便可)裝葉子節點。
而後寬度優先搜索隊列中葉子節點，刪除節點和它們對應的邊，並加入新的葉子節點(有可能刪除葉子節點的邊以後，它的鄰接點p的入度變爲1,也就是inDegree[p] = 1，此時能夠把p加入隊列中)。
直到只剩下1個或者2個點中止。返回結果。

java ac代碼：
class Solution {
    public List<Integer> findMinHeightTrees(int n, int[][] edges) {
    
        List<Integer>ans = null;
        if(n <= 2)
        {
            ans = new ArrayList(2);
            for(int i = 0;i < n;i++)
                ans.add(i);
            return ans;
        }
        
        //去掉葉子節點，直到只剩下一個或者兩個節點。
        int[]inDegree = new int[n];
        
        //用鄰接表存儲圖。
        List<Integer>[] map = new List[n];
        for(int i = 0;i < n;i++)
            map[i] = new ArrayList();
       for(int i = 0;i < edges.length;i++)
       {
           map[edges[i][0]].add(edges[i][1]);
           map[edges[i][1]].add(edges[i][0]);
           inDegree[edges[i][0]]++;
           inDegree[edges[i][1]]++;
       }
        
        int count = n;
        Queue<Integer>queue = new ArrayDeque();
            for(int i = 0;i < n;i++)
                if(inDegree[i] == 1)
                    queue.offer(i);
        while(count > 2)
        {
            int size = queue.size();
            for(int loc  = 0;loc < size;loc++)
            {
                int ii = queue.poll();
                int jj = map[ii].get(0);
               
                inDegree[jj]--;
                if(inDegree[jj] == 1)
                    queue.offer(jj);
                
                map[jj].remove(new Integer(ii));
                map[ii].remove(new Integer(jj));
                inDegree[ii] = -1;
                
                count--;
            }
        }
    
        ans = new ArrayList(queue);
        
        return ans;
    }
    
    
}

這個和力扣(LeetCode)207很像，雖然不是拓撲排序，但都是利用一個可彈出容器裝節點，而後寬度搜索容器中的節點。