hdu 4006 The kth great number （优先队列）

优先队列头文件

#include <queue>

默认优先为从大到小优先。

自定义优先级

struct cmpmin{                         //按从小到大         

// 因为标准库默认使用元素类型的<操作符来确定它们之间的优先级关系。
    bool operator ()(int &a, int &b){
        return a>b;                       //所以规定小的元素的优先级大于大的元素。 
    }
};

struct cmpmax{                        //按从大到小          
    bool operator ()(int &a, int &b){
        return a<b;                       
    }
};


//多个元素的时候 
struct node{      
    int x, y;
    friend bool operator <(int a, int b){        //小的优先级小，所以按照x从大到小，注意默认是<来确定优先级。 
        return a.x < b.x;
    }
}

struct node{
    int x, y;
    friend bool operator <(int a, int b){        //小的优先级大，所以按照x从小到大 
        return a.x > b.x;                       
    }
}

 

本题要输出第K大的数，因为数据量大，不能每次排序，所以用优先队列来解决。

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
using namespace std;
struct cmp{
    bool operator ()(int &a,int &b){
        return a>b;//最小值优先
    }
};
int n, k, val, cnt;
char c;
int main(){
    while(~scanf("%d%d", &n, &k)){
        cnt = 0;
        priority_queue <int, vector<int>,cmp> q;
        while(n--){
            cin>>c;
            if(c == 'Q'){
                printf("%d\n", q.top());
            }
            else if(c == 'I'){
                scanf("%d", &val);
                if(cnt < k){
                    q.push(val);cnt++;
                }
                else{
                    if(val > q.top()){
                        q.pop();
                        q.push(val);
                    }
                }
            }
        }
        
    }
    
    
    return 0;
}