hdu4006 The kth great number（优先队列）

hdu4006

题目

有插入和询问操作，插入的是数字，询问的是第k大的数是几。

思路

用优先队列维护一个有k个数的集合，未满k时入队，满k时判断是不是比最小的要大，大则进去，小的出来，这样查询的时候最小的就是第k大的了。相当于用较小的代价保住了我们需要的部分而且不出错。

代码

#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>

using namespace std;

int main()
{
    int n,k;
    char s[20];
    int temp;
    int cnt;
    while(scanf("%d %d",&n,&k)!=EOF)
    {
        priority_queue<int,vector<int>,greater<int> > q;
        while(!q.empty()) q.pop();
        cnt=0;

        for(int i=0; i<n; i++)
        {
            scanf("%s",s);
            if(s[0]=='I')
            {
                scanf("%d",&temp);
                if(cnt<k)
                {
                    q.push(temp);
                    cnt++;
                }
                else
                {
                    if(temp>q.top())
                    {
                        q.pop();
                        q.push(temp);
                    }
                }
            }
            else if(s[0]=='Q')
            {
                printf("%d\n",q.top());
            }
        }
    }
    return 0;
}