【洛谷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=,f=; char c=getchar();
     while(c<''||c>'') { if(c=='-') f=-; c=getchar(); }
     while(''<=c&&c<='') { x=(x<<)+(x<<)+c-''; c=getchar(); }
     return x*f;
 }
 int main()
 {
     scanf("%lld%lld",&m,&n);
     for(LL i=;i<=m;i++)
      a[i].data=read();
     for(LL i=;i<=n;i++)
      a[read()].get++;
     h1.push(-INF); h2.push(INF);
     for(LL i=;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=;j<=a[i].get;j++){
             LL t2=h2.top();
             h2.pop();
             h1.push(t2);
             printf("%lld\n",h1.top());
         }
     }
     return ;
 }