剑指offer 29.最小的K个数

剑指offer 29.最小的K个数

题目

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

思路

几种方法,我一开始想到的是直接快排函数,然后输出k个就行,之后改用优先队列建堆,大小控制为k,大于这个数量就删除堆顶(最大值),最后只剩k个,输出到ans当中即可。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {

ArrayList<Integer> ans = new ArrayList<>();
if (input == null || k <= 0 || k > input.length) {
return ans;
}
Queue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2 - o1);

for (int i = 0; i < input.length; i++) {

if (queue.size() < k) {
queue.add(input[i]);
} else {
if (input[i] < queue.peek()) {
queue.remove();
queue.add(input[i]);
}
}
}
while (!queue.isEmpty()) {
ans.add(queue.remove());
}
return ans;
}
---本文结束,感谢阅读---