编程练习【最小k个数】

设计一个算法，找出数组中最小的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>() {
			@Override
			public 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;
    }