HDU - 4006 The kth great number 优先队列最小堆

HDU - 4006 The kth great number

题意：

给你n条指令（两种操作，" I "插入元素，“Q”输出第k个大的数）。第K个大的数是指把这些数降序排列后，从大到小的第K个数。一开始理解反了，看了案例郁闷了半天。。。

思路：

因为K是不变的，所以我们只需让 优先队列实现的最小堆里 保存K个元素就行。当元素少于K个时，直接插入。当元素大于K个时，将输入的元素与队顶元素进行比较，如果队顶元素小于输入的元素就将对顶元素出队，把此元素入队，否则不变。这样查询输出时直接输出队顶元素即为答案。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
priority_queue<int, vector<int>,greater<int> > heap;
int main()
{
    int n,k;
    char s;
    int x;
    while(~scanf("%d %d",&n,&k))
    {
        while(!heap.empty())
            heap.pop();
        for(int i=0;i<n;i++)
        {
            scanf(" %c",&s);
            if(s=='I')
            {
                scanf("%d",&x);
                if(heap.size() < k)   //不到K个
                    heap.push(x);
                else
                {
                    int a = heap.top();
                    if(a < x)     //交换队顶元素
                    {
                        heap.pop();
                        heap.push(x);
                    }
                }
            }
            else
                printf("%d\n",heap.top());
        }
    }
    return 0;
}