/* 輸入 n 個整數,輸出其中最小的 k 個 * TreeMap 和 TreeSet 是 Java Collection Framework 的兩個重要成員, * 其中 TreeMap 是 Map 接口的經常使用實現類,而 TreeSet 是 Set 接口的經常使用實現類 * 。雖然 HashMap 和 HashSet 實現的接口規範不一樣,但 TreeSet 底層是經過 TreeMap 來實現的,所以兩者的實現方式徹底同樣。而 TreeMap 的實現就是紅黑樹算法。 * 1. 紅黑樹並不追求「徹底平衡」——它只要求部分地達到平衡要求,下降了對旋轉的要求,從而提升了性能 * 2.AVL樹是最早發明的自平衡二叉查 找樹。在AVL樹中任何節點的兩個兒子子樹的高度最大差異爲一,因此它也被稱爲高度平衡樹。查找、插入和刪除在平均和最壞狀況下都是O(log n)。增長和刪除可能須要經過一次或屢次樹旋轉來從新平衡這個樹。 * 3. MyComparator 不能夠返回爲0, 爲0表示相等,TreeSet 裏面只存惟一的數字 * 由於咱們使用 new MyComparator() 修改了她的默認排序規則 */ public static void findkmin(int[] array, int k){ TreeSet<Integer> kminset = new TreeSet<Integer>(); for(int i = 0 ; i < array.length; i ++){ if(kminset.size() < k ){ kminset.add(array[i]); continue; } if(array[i] < kminset.first()){ kminset.remove(kminset.last().intValue()); // 假如 TreeSet<Integer> kminset = new TreeSet<Integer>(new MyComparator()); 是這樣申請的 // treesset 刪除一個元素的時候必須刪除最後一個,由於刪除開始或者中間的一個會引發其排序不穩定,因此刪除會失敗 // 由於咱們使用 new MyComparator() 修改了她的默認排序規則 kminset.add(array[i]); } } System.out.println("***************************************************"); for(Integer in : kminset){ System.out.print(in +" "); } System.out.println("\n***************************************************"); }