那天在写东西的时候发现好像priority_queue是没办法直接构建大小的,想想也是,一个容器适配器怎么去给他固定size呢,然后看了一下priority_queue的构造函数确实也是不能这么搞的
同样的对于stack,queue这些控制大小都可以按下面思路做
那怎么办呢,比如给个固定大小k,让你找到第k个最大的元素。
没法直接构建,那就在插入元素循环里面if控制一下就行
如下:
int findKthLargest(vector<int>& nums, int k) {
priority_queue<int,vector<int>,greater<>>sq;
for(auto &x:nums) //让sq里面始终保持k个元素
{
if(sq.size()<k)
{
sq.push(x);
}
else if(sq.top()<x){
sq.pop();
sq.push(x);
}
}
return sq.top();
}