题目[1]
最小的k个数
难度简单351收藏分享切换为英文接收动态反馈
输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。
示例 1：
输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]
示例 2：
输入：arr = [0,1,2,1], k = 1
输出：[0]
解答
新手上路，才学疏浅，望斧正
public class Solution17_1 {
public int[] getLeastNumbers(int[] arr, int k) {
if(k==0){
return new int[k];
}
PriorityQueue<Integer> queue=new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2-o1;
}
});
for (int i=0; i<k; i++){
queue.add(arr[i]);
}
for(int i=k;i<arr.length;i++){
if(queue.peek()>arr[i]){
queue.poll();
queue.add(arr[i]);
}
}
int[] res=new int[k];
for (int i=0;i<k;i++){
res[i]=queue.poll();
}
return res;
}
}