题目
输入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;}};
