设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。
示例:
输入: arr = [1,3,5,7,2,4,6,8], k = 4
输出: [1,2,3,4]
题目来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/smallest-k-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法一:排序复制
将数组排序,再将数组按顺序复制k个到结果数组中。
代码:
public static int[] smallestK(int[] arr, int k) {Arrays.sort(arr);int[] ret=new int[k];for(int i=0;i<k;i++) {ret[i]=arr[i];}return ret;}
方法二:队列
利用优先队列的特点,将数组前k个数字先放入队列中,然后遍历剩下的数字,和队列第一个元素比较,如果队列第一个元素大于后面的数字,则数字出队,小的数字入队,遍历完数组后即可得到最小k个数的一个队列,再将元素保存到数组中即可。
代码:
public static int[] smallestK(int[] arr, int k) {int[] ret=new int[k];if(k==0) {return ret;}PriorityQueue<Integer> queue=new PriorityQueue<Integer>(new Comparator<Integer>() {@Overridepublic int compare(Integer o1, Integer o2) {return o2-o1;//定义比较规则}});for(int i=0;i<k;++i) {queue.offer(arr[i]);}for(int i=k;i<arr.length;++i) {if(queue.peek()>arr[i]) {queue.poll();queue.offer(arr[i]);}}for(int i=k-1;i>=0;i--) {ret[i]=queue.poll();}return ret;}