【暖*墟】#对顶堆# 中位数的动态维护 poj3784

【对顶堆：中位数的动态维护】

#include <bits/stdc++.h> //貌似poj不能用这个...
using namespace std;
typedef long long ll;

/*【Running Median】poj3784
输入n个数，对第k个输入，如果k为奇数，则输出前k个数的中位数。 */

// 动态求中位数。运用（对顶堆）即一个大顶堆和一个小顶堆。
// 每次输入，如果这个数比当前的中位数大，就存入小顶堆中，否则就存入大顶堆。
// 调整：小顶堆元素的个数要等于大顶堆的元素个数，或者比其多1。 
// 如果小顶堆的元素太多，就塞到大顶堆里，反之亦然。
// 发现：小顶堆元素比所有大顶堆元素都大，而且小顶堆的堆顶就是中位数。 

priority_queue<int, vector<int>, greater<int> > q1; //小顶堆
priority_queue<int, vector<int>, less<int> > q2; //大顶堆

vector<int> g; //用于输出

void add(int x) {
    if(q1.empty()) { //一开始，先输入小顶堆
        q1.push(x); return;
    }
    if(x > q1.top()) q1.push(x);
    else q2.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();
    }
}

int main() {
    int T, cas, n, x;
    scanf("%d", &T);
    while(T--) {
        while(!q1.empty()) q1.pop(); //清空堆
        while(!q2.empty()) q2.pop();
        g.clear(); //初始化清零
        scanf("%d%d", &cas, &n);
        for(int i = 0; i < n; i++) {
            scanf("%d", &x); add(x);
            if(i % 2 == 0) g.push_back(q1.top()); 
        } //↑↑↑奇数情况（因为从0开始，所以变成了偶数）
        printf("%d %d\n", cas, (n + 1) / 2); //编号、输出中位数的次数
        for(int i = 0; i < g.size(); i++) {
            if(i > 0 && i % 10 == 0) putchar('\n');
            if(i % 10) putchar(' ');
            printf("%d", g[i]);
        }
        printf("\n");
    }
    return 0;
}