文章目录
1. 题目信息
1.1 题目描述
题目链接: 面试题 17.14. 最小K个数
设计一个算法,找出数组中最小的k个数。以 任意顺序
返回这 k
个数均可。
1.2 测试用例
- 示例:
输入: 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))
2. 题目分析
2.1 sort后取前k个
很直观的思路, 排序后, 取 最小的 K 个
2.2 大根堆 存最小的K个
C++中的大根堆, 就是
priority_queue
, 默认top
存放最大元素;
- 先存 前面 K 个元素 到 堆中;
- 遇到新的数字和堆里面的 最大值即 top 进行比较, 如果当前值更小, 则 入堆, 替换掉原来的最大值。
- 最后剩下的元素放入数组中返回即可;
3. 代码详情
3.1 C++
3.1.1 sort方法
class Solution {
public:
vector<int> smallestK_sort(vector<int>& arr, int k) {
sort(arr.begin(), arr.end());
return vector<int>(arr.begin(), arr.begin() + k);
}
};
3.1.2 大根堆方法
class Solution {
public:
vector<int> smallestK(vector<int>& arr, int k) {
// 执行用时:36 ms, 在所有 C++ 提交中击败了46.77%的用户
// 内存消耗:18.2 MB, 在所有 C++ 提交中击败了37.65%的用户
if (k == 0) {
return vector<int>{};
}
priority_queue<int> small;
for (int i : arr) {
if (small.size() < k) {
small.push(i);
} else {
if (i < small.top()) {
small.pop();
small.push(i);
}
}
}
vector<int>ans(k);
int cnt = 0;
while(big.size()) {
ans[cnt++] = big.top();
big.pop();
}
return ans;
}
};
3.2 Python
- 略