题目
输入n个整数,找出其中最小的k个数。
注意:
数据保证k一定小于等于输入数组的长度;
输出数组内元素请按从小到大顺序排序;
样例
输入:[1,2,3,4,5,6,7,8] , k=4
输出:[1,2,3,4]
解法:优先队列
维护大小为k的优先队列,然后存入vector中倒序
时间复杂度O(nlogk),空间复杂度O(k)
class Solution {
public:
priority_queue<int> q;
vector<int> getLeastNumbers_Solution(vector<int> input, int k) {
for (auto x: input) {
if (q.size() >= k) {
if (x < q.top()) {
q.pop();
q.push(x);
}
}
else q.push(x);
}
vector<int> ans;
while (q.size()) {
ans.push_back(q.top());
q.pop();
}
reverse(ans.begin(), ans.end());
return ans;
}
};