[P1801]黑匣子

原题链接

思路来自题解

使用两个堆来维护
大根堆
记录当前
前s个最小的数

多出来的数
扔到小根堆里
等需要输出的时候
把小根堆的顶
输出
然后删掉
放进大根堆里

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;

priority_queue<int>a;//大根堆 
priority_queue<int>b;//小根堆 

int m,n,num[200000+5],u[200000+5],i,s,k,ans;

int main()
{
    scanf("%d%d",&m,&n);

    for(i=1;i<=m;i++)
        scanf("%d",&num[i]);        

    for(i=1;i<=n;i++)
        scanf("%d",&u[i]);

    s=0;
    k=1;

    for(i=1;i<=m;i++)
    {
        a.push(num[i]);

        if(a.size()>s)
        {

            b.push(-a.top());
            a.pop();            
        } 

        while(u[k]==i)
        {
            s++;            

            ans=-b.top();
            a.push(ans);
            b.pop();

            printf("%d\n",ans);

            k++;
        }
    }

    return 0;   
}