HDU 4006 第K大元素（小技巧）

题意

输入一些数，进行多次查询。每次查询都是查询第K大元素是什么。

题解

这道题考察的是堆（优先队列）的应用。。。。核心代码10行，主要是考察思想。既然每次都是查询第K大元素，那么就做一个最小堆，大小为K。每次增加元素都尝试放入这个最小堆，如果新增加的元素比这个最小堆的最小值要大，则把新元素插入最小堆，并将移除原来的最小值。这样的话，每次取出来的最小值就是第K大的元素。

额外的尝试

c++ STL 有个nth-element，利用类似快排的思想，查找第N个小的数，并把比他小的放在前面，比他大的放在后面（随机顺序）。很可惜，这种算法效率还不够高，TLE。。。

代码

#include <iostream>#include<cstdio>#include<algorithm>#include<cstring>#include<vector>#include<cmath>#include<queue>#include<string>#include<set>#include<map>#define UP(i,l,h) for(int i=l;i<h;i++)#define DOWN(i,h,l) for(int i=h;i>l;i--)#define W(a) while(a)#define INF 0x3f3f3f3fusing namespace std;priority_queue<int,vector<int>,greater<int>> que;int main() {    int n,k;    while(~scanf("%d%d",&n,&k)) {        while(!que.empty()){            que.pop();        }        int pos=0;        W(n--) {            char ch[10];            scanf("%s",ch);            if(strcmp(ch,"I")==0) {                int x;                scanf("%d",&x);                if(que.size()<k) {                    que.push(x);                } else if(que.top()<x){                    que.pop();                    que.push(x);                }            }else{                printf("%d\n",que.top());            }        }    }    return 0;}