最开始我的想法是把所有数据都保存在优先队列里,然后每当弹出时就用其他容器接住,然后再放回优先队列中,结果果不其然超时了。
于是看了题解,觉得有道理,可以用大根堆(less)来保存,先将所有数据保存在堆中,如果堆的大小大于k,那么也就是说后面的数据压根不可能成为答案,所以就出队。然后接着进行选择,如果当前是插入,继续进行之前判断,如果当前堆的大小大于k,弹出
核心就是让第k小的数成为堆顶
#include<iostream>
#include<queue>
using namespace std;
int n,m,k;
priority_queue<int,vector<int>,less<int>> q;
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
q.push(x);
}
while(q.size()>k)q.pop();
while(m--)
{
int choice;
cin>>choice;
if(choice==1)
{
int x;
cin>>x;
q.push(x);
if(q.size()>k)q.pop();
}
else
{
if(k>q.size())cout<<-1<<endl;
else cout<<q.top()<<endl;
}
}
return 0;
}