直接维护一个长度为K的优先队列,即可。
这题学习了优先队列的另一种定义方法:
#include <algorithm> #include <vector> #include <queue> #include <cstdio> using namespace std; int main() { int n,k; while(~scanf("%d%d",&n,&k)) { priority_queue< int,vector<int>,greater<int> >que;//小顶堆 while(n--){ char op[5]; scanf("%s",op); if(op[0]=='I') { int val; scanf("%d",&val); que.push(val); while(que.size()>k) que.pop(); } else printf("%d\n",que.top()); } } return 0; }
下面拓展一下优先队列的知识:
一般优先队列的定义方法:
priority_queue< Type,vector<Type>,greater<Type> >(小顶堆)
priority_queue< Type,vector<Type>,less<Type> >(大顶堆)
如果是自定义的结构体类型
priority_queue<Node,vector<Node>,cmp>
需要自己自定义结构体类型:
struct cmp
{
bool operator()(const Node &t1,const Node &t2)
{
return t1.b<t2.b;//相当于less,大顶堆
}
};