[HDOJ] problem_4006 【The kth great number】 优先队列

[HDOJ] problem_4006 【The kth great number】：

点击打开链接

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;

bool cmp(int a,int b)
{
    return a>b;
}

int main()
{
	
    int i,j,k,n,a,r;
    char c;
    while (scanf("%d %d",&n,&k)!=-1)
    {
		priority_queue<int,vector<int>,greater<int> > q;   //这里最后> >之间一定要有个空格，否则会出错，看了网上一些例子才发现。。虽然还不知道为什么
        r=0;
        while (n--)
        {
            getchar();
            c=getchar();
            if (c=='I')
            {
				scanf("%d",&a);
				if (q.size()==k)
				{
					if (q.top()<a)
					{
						q.pop();
						q.push(a);
					}
				}
				else
					q.push(a);
            }
            else if (c=='Q')
            {
                printf("%d\n",q.top());
            }
        }
    }
    return 0;
}

这题要是直接排序会超时，不久前学了优先队列，哈哈，果然多学点东西没错。

这题用最小堆做，将元素放入优先队列中，当队列里元素数量满足k时，比较新的元素与队头元素的大小，如果比队头元素大，则删除队头元素，放入新的元素；若小，判断下一个新的元素。这样可保证队头元素永远是第k大的数。