[NOI导刊2010提高]黑匣子

OJ题号：洛谷1801

思路：建立一个大根堆、一个小根堆。大根堆维护前i小的元素，小根堆维护当前剩下的元素。

#include<cstdio>
#include<queue>
#include<functional>
#include<vector>
int main() {
    int m,n;
    scanf("%d%d",&m,&n);
    int a[m],u[n+1];
    for(int i=0;i<m;i++) scanf("%d",&a[i]);
    u[0]=0;
    for(int i=1;i<=n;i++) scanf("%d",&u[i]);
    std::priority_queue<int,std::vector<int>,std::less<int> > b;
    std::priority_queue<int,std::vector<int>,std::greater<int> > s;
    unsigned int p=0;
    for(int i=1;i<=n;i++) {
        p++;
        while((b.size()<p)&&!s.empty()) {
            b.push(s.top());
            s.pop();
        }
        for(int j=u[i-1];j<u[i];j++) {
            if(b.size()<p) {
                b.push(a[j]);
            }
            else {
                if(a[j]<b.top()) {
                    s.push(b.top());
                    b.pop();
                    b.push(a[j]);
                }
                else {
                    s.push(a[j]);
                }
            }
        }
        printf("%d
",b.top());
    }
    return 0;
}