P1484 种树

Posted 2020-12-18 for-miku

链接：Miku

---------------------------------

对顶堆做法

---------------------------------

#include<iostream> 
#include<cstdio>
#include<algorithm>
#include<queue>

using namespace std;
 priority_queue <int,vector<int>,greater<int> > q1;//小顶 
 priority_queue <int,vector<int>,less<int> >q2;//大顶 
 int n,k;
 int x;
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%d",&k);
        if(i==1){
        q2.push(k);
        cout<<k<<endl;
        continue;
        }
        else{
            if(q2.top()>k)
            q2.push(k);
            else
            q1.push(k);
        }
        if(q2.size()>q1.size()&&q2.size()-q1.size()>1){//不先比较会有莫名其妙越界问题
            x=q2.top();
            q1.push(x);
            q2.pop();
        }
        if(q1.size()>q2.size()&&q1.size()-q2.size()>1){
            x=q1.top();
            q2.push(x);
            q1.pop();    
        }
        if(i%2){
            if(q2.size()>q1.size())
            cout<<q2.top()<<endl;
            else
            cout<<q1.top()<<endl;
        }
    }
    return 0;
}

Ac