【洛谷P1801】黑匣子

黑匣子

题目链接

看到题解中“维护两个堆”，突然想到了这道题的解法

维护两个堆：大根堆h1, 小根堆h2

大根堆里的是最小的i个值，小根堆里是剩下的值

每Add一个值时

插入到小根堆中，

再比较小根堆的最小值与大根堆的最大值

若h2.top()<h1.top()

将两个元素取出，换一下再放进去

需要Get时

将h2.top()取出，放进h1中，

再输出h1.top()

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
#define LL long long
#define N 200010
#define INF 1e18
LL m,n;
struct Add{
    LL data;
    LL get;
} a[N];
struct cmp{
    bool operator ()(LL x,LL y){
        return x>y;
    }
};
priority_queue<LL> h1;
priority_queue< LL, vector<LL>, cmp > h2;
inline LL read(){
    LL x=0,f=1; char c=getchar();
    while(c<'0'||c>'9') { if(c=='-') f=-1; c=getchar(); }
    while('0'<=c&&c<='9') { x=(x<<3)+(x<<1)+c-'0'; c=getchar(); }
    return x*f;
}
int main()
{
    scanf("%lld%lld",&m,&n);
    for(LL i=1;i<=m;i++)
     a[i].data=read();
    for(LL i=1;i<=n;i++)
     a[read()].get++;
    h1.push(-INF); h2.push(INF);
    for(LL i=1;i<=m;i++){
        h2.push(a[i].data);
        if(h1.top()>h2.top()){
            LL t1=h1.top(),t2=h2.top();
            h1.pop(); h2.pop();
            h1.push(t2); h2.push((t1));
        }
        for(LL j=1;j<=a[i].get;j++){
            LL t2=h2.top();
            h2.pop();
            h1.push(t2);
            printf("%lld
",h1.top());
        }
    }
    return 0;
}