题目
代码
class Solution {
public int[] getLeastNumbers(int[] arr, int k) {
if (k == 0 || arr.length == 0) return new int[0];
//实现升序排列即头节点是最大的--需要重写一下比较器。
Queue<Integer> pq = new PriorityQueue<>((v1, v2) -> v2-v1);
for (int num: arr) {
if (pq.size() < k) pq.offer(num);
else if (num < pq.peek()) {
pq.poll();
pq.offer(num);
}
}
// 返回堆中的元素
int[] res = new int[pq.size()];
int idx = 0;
for(int num: pq) { res[idx++] = num; }
return res;
}
}