1,问题简述
输入整数数组 arr ,找出其中最小的 k 个数。
例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
2,示例
代码语言:javascript复制示例 1:
输入:arr = [3,2,1], k = 2
输出:[1,2] 或者 [2,1]
示例 2:
输入:arr = [0,1,2,1], k = 1
输出:[0]
限制:
0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000
3,题解思路
使用排序,队列的方式进行解决
4,题解程序
代码语言:javascript复制
import java.util.Arrays;
import java.util.Collections;
import java.util.PriorityQueue;
import java.util.Queue;
public class GetLeastNumbersTest {
public static void main(String[] args) {
int[] arr = {4, 5, 1, 6, 2, 7, 3, 8};
int k = 4;
int[] leastNumbers = getLeastNumbers2(arr, k);
for (int num : leastNumbers
) {
System.out.print(num "t");
}
}
public static int[] getLeastNumbers(int[] arr, int k) {
if (arr == null || k == 0) {
return new int[0];
}
Queue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2 - o1);
for (int num : arr) {
if (queue.size() < k) {
queue.offer(num);
} else if (queue.peek() > num) {
queue.poll();
queue.offer(num);
}
}
int[] result = new int[k];
for (int i = 0; i < k; i ) {
result[i] = queue.poll();
}
return result;
}
public static int[] getLeastNumbers2(int[] arr, int k) {
if (arr == null || arr.length < k) {
return new int[0];
}
Arrays.sort(arr);
int[] result = new int[k];
if (k >= 0) {
System.arraycopy(arr, 0, result, 0, k);
}
return result;
}
}
5,题解程序图片版
6,总结一下
这道题给出了两个题解思路,两种方式都是比较容易理解的,这里就没有给出过多的注释说明,队列的特点在之前的文章中多次说明了,就是先进先出,但是对于应用的开发者来说,熟悉队列的特点也是有必要的,这里就简单的说下