设计一个算法,找出数组中最小的k个数,以任意顺序返回这k个数均可
class Solution {
public:
vector<int> smallestK(vector<int>& arr, int k) {
vector<int>res;
priority_queue<int, vector<int>, greater<int>>q;
for (int i = 0; i < arr.size(); i++) {
q.push(arr[i]);
}
while(k--) {
res.push_back(q.top());
q.pop();
}
return res;
}
};
priority_quene 实现优先队列的结构,优先级高的元素先出队列,有三个元素,
priority_quene<Type,Container,Functional>
Type:类型数据
Container:保存数据的容器,比如vector和deque,而不能是list
Functional:元素的比较方式,有less和greater,默认是大顶堆,队头元素最大
priority_queue<结构类型> 队列名;
priority_queue<int,vector,less >q; // 从大到小,默认的
priority_queue<int,vector,greater >q; // 从小到大,队列前面是小,后面是大