用小根堆来保存k大的元素,堆顶即为第k大的元素。
class KthLargest {
priority_queue<int, vector<int>, greater<int> > pq;
int n;
public:
KthLargest(int k, vector<int>& nums) {
n = k;
for(auto it: nums){
add(it);
}
}
int add(int val) {
if(pq.size() < n) pq.push(val);
else if(pq.top() < val){
pq.pop();
pq.push(val);
}
return pq.top();
}
};
优先队列
常用函数 push() pop() top() size() empty()
优先队列的本质是堆
#include<bits/stdc++.h>
using namespace std;
int main(){
priority_queue<int> q; //默认大根堆
priority_queue<int, vector<int>, greater<int> > p; //小根堆
p.push(3);
p.push(6);
p.push(35);
q.push(3);
q.push(6);
q.push(35);
cout<<q.top()<<endl; //35
cout<<p.top()<<endl; //3
return 0;
}