用小根堆来保存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;
}