【POJ3784】Running Median（中位數，對頂堆）

description

輸入M個數，當已輸入的個數為奇數個時輸出此時的中位數。
一共有M/2+1箇中位數要輸出，每一行10個。

solution

維護兩個優先佇列。
大根堆q1維護比當前中位數小的元素。
小跟堆q2維護比當前中位數大的元素。

我們把中位數放在大根堆q1的堆頂，於是有：
輸入的數為奇數個時：q1.size()=q2.size()+1 （多一箇中位數）
輸入的數為偶數個時：q1.size()=q2.size() （不存在中位數）

codes

#include<iostream>
#include<queue>
using namespace std;
priority_queue<int>q1;//big
priority_queue<int,vector<int>,greater<int> >q2;
int main(){
    int T;  cin>>T;
    while(T--){
        while(q1.size())q1.pop();
        while(q2.size())q2.pop();

        int cas, n;
        cin>>cas>>n;
        cout<<cas<<' '<<(n+1)/2<<'\n';
        for(int i = 1; i <= n; i++){
            int x;  cin>>x;

            if(q1.empty())q1.push(x);
            else{
                if(x>q1.top())q2.push(x);
                else q1.push(x);
            }

            while(q1.size()<q2.size()){
                q1.push(q2.top());  q2.pop();
            }
            while(q1.size()>q2.size()+1){
                q2.push(q1.top());  q1.pop();
            }

            if(i&1)cout<<q1.top()<<' ';
            if(!(i%20))cout<<'\n';
        }

        cout<<'\n';
    }
}