P1801 黑匣子

题面:https://www.luogu.org/problemnew/show/P1801

本题就是把元素先都放入大根堆,然后当大根堆中元素大于等于k个时,则把多余元素放入小根堆中,这样保证小根堆堆顶是第k小的
然后当输出完第k小的数后,则要把它放回大根堆(因为k增加了)

Code:
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
int a[200005],g[200005];
priority_queue<int> B;
priority_queue<int,vector<int>,greater<int> > A;
int main(){
    int n,m,k=1;
    scanf("%d%d",&n,&m);
    for(int x=1;x<=n;x++){
        scanf("%d",&a[x]);
	}
    for(int x=1;x<=m;x++){
        scanf("%d",&g[x]);
	}
    for(int j=1;j<=n;j++){
        B.push(a[j]);
        if(B.size()>=k){
			A.push(B.top());
			B.pop();
		}
        while(j==g[k]){
            printf("%d\n",A.top());
            B.push(A.top());
            A.pop();
            k++;
        }
    }
    return 0;
}