Java实现最小高度树_java

class Solution {      public List<Integer> findMinHeightTrees(int n, int[][] edges) {          List<Integer> res = new ArrayList<Integer>();          //如果只有一个节点，则它就是最小高度树          if(n == 1){              res.add(0);              return res;          }            //每个节点的邻居数量          int [] degree = new int[n];          //每个节点的邻居          HashMap<Integer,List<Integer>> map = new HashMap<>();            for(int [] edge : edges){              int a = edge[0];              int b = edge[1];                degree[a]++;              degree[b]++;                if(map.get(a) == null){                  map.put(a,new ArrayList<Integer>());//key:节点   value:邻居              }                if(map.get(b) == null){                  map.put(b,new ArrayList<Integer>());//key:节点   value:邻居              }                map.get(a).add(b);              map.get(b).add(a);          }            //建立队列          LinkedList<Integer> leafNodes = new LinkedList<Integer>();//表示叶子节点          //将所有度为1的节点入队          for(int i = 0;i < degree.length;i++){              if(degree[i] == 1){                  leafNodes.add(i);              }          }            while(leafNodes.size() > 0){              res.clear();              //每一层节点的数量              int size = leafNodes.size();                for(int i = 0;i < size;i++){                  int leaf = leafNodes.poll();                  //将当前节点加入到结果集                  res.add(leaf);                    List<Integer> neighbors = map.get(leaf);                  //将出度减一，也就是将最外层的叶子节点剪掉                  for(int neighbor : neighbors){                      degree[neighbor]--;                      if(degree[neighbor] == 1){                          //叶子节点入队                          leafNodes.add(neighbor);                      }                  }              }          }          return res;      }  }