【luoguP1168】中位数

题目链接

用一个大根堆和一个小根堆维护中位数即可

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;

int n,sz1,sz2;

priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int> > q2;

int main(){
    scanf("%d",&n);
    int x;
    for(int i=1;i<=n;++i){
        scanf("%d",&x);
        if(!q1.empty()&&x>q1.top())
            q2.push(x),++sz2;
        else q1.push(x),++sz1;
        while(sz2+1<sz1){
            q2.push(q1.top());
            q1.pop();
            ++sz2; --sz1;
        }
        while(sz1<sz2){
            q1.push(q2.top());
            q2.pop();
            ++sz1; --sz2;
        }
        if(i%2==1) printf("%d\n",q1.top());
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/yjkhhh/p/11833918.html