力扣面试题 17.14. 最小K个数
设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。
示例:
输入: arr = [1,3,5,7,2,4,6,8], k = 4
输出: [1,2,3,4]
提示:数组内的数输出顺序可以不同。
0 <= len(arr) <= 100000
0 <= k <= min(100000, len(arr))
- 使用堆排序
使用原数组的前k个元素创建一个最大堆,剩余元素与最大堆的首元素比较,若较之更小,则与之替换,遍历后就可以得到原数组中的最小k个数。
代码如下:
class Solution {
public:
vector<int> smallestK(vector<int>& arr, int k) {
vector<int> vec;
if (k == 0)
return vec;
vec.resize(k);
int maxindex = k - 1, curindex = (maxindex - 1) / 2,n = arr.size();
for (int i = k; i < n; i++)
{
for (int j = curindex; j >= 0

这篇博客介绍了如何在C++中找到数组中最小的k个数,提供了两种算法实现:堆排序和快速排序。堆排序通过建立最大堆并不断调整来找到最小k个数;快速排序则利用荷兰三色旗思想进行数组划分。两种方法都确保了输出顺序的任意性,并满足了给定的时间和空间复杂度要求。
最低0.47元/天 解锁文章
2227

被折叠的 条评论
为什么被折叠?



