技術標籤：演算法練習演算法

題目描述(劍指Offer40)

輸入n個整數，找出其中最小的K個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4。

示例1
輸入

[4,5,1,6,2,7,3,8],4

返回值

[1,2,3,4]

public class Jzof40 {

    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> arrayList = new ArrayList
 
<>();
        if (k > input.length) return arrayList;
        Arrays.sort(input);
        for (int i = 0; i < k; i++) {
            arrayList.add(input[i]);
        }
        return arrayList;
    }
    /**********使用堆************/
    /**********別人的解法********/
    public ArrayList<Integer>
 
 GetLeastNumbers_Solution1(int [] input, int k) {
        if (k > input.length || k < 0){
            return new ArrayList<>();
        }
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(((o1, o2) -> o2 - o1));
        for (int i : input) {
            maxHeap.add
 
(i);
            if (maxHeap.size() > k){
                //獲取並刪除隊首元素
                maxHeap.poll();
            }
        }
       return new ArrayList<>(maxHeap);
    }
}

Java 的 PriorityQueue 實現了堆的能力，PriorityQueue 預設是小頂堆，可以在在初始化時使用 Lambda 表示式 (o1, o2) -> o2 - o1 來實現大頂堆