poj 3784 Running Median 二叉堆

传送门

---

题目大意：
依次添加 n 个数，要求第奇数次添加时，输出当前数列中的中位数。

---

分析：
维护一个大根堆一个小根堆，大的数插入小根堆，小的数插入大根堆，每次取大根堆堆顶就是中位数

---

代码如下：（第一次手写二叉堆，所以是抄的POPOQQQ的模板>o< >_<）

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=5000+5;
int p,a,m;
struct small_root_heap{//小根堆 
    int heap[maxn],size;
    void insert(int x){
        heap[++size]=x;
        int t=size;
        while(t>1&&heap[t]<heap[t>>1])//不断向上调整直到比父结点不优为止
            swap(heap[t],heap[t>>1]),t>>=1;
    }
    void pop(){//弹顶 
        int t=2;
        heap[1]=heap[size],heap[size--]=0;
        while(t<=size){
            if(heap[t]>heap[t+1]&&t<size)
                t++;
            if(heap[t]<heap[t>>1])//不断向上调整直到比父结点不优为止
                swap(heap[t],heap[t>>1]),t<<=1;
            else
                break;
        }
    }
}topheap,bottomheap,empty;//topheap是小根堆，bottomheap是大根堆
void add(int x){
    if(x<=-bottomheap.heap[1])
        bottomheap.insert(-x);
    else
        topheap.insert(x);//小的插入大根堆，大的插入小根堆 
    //插入，保证下堆的任意元素都小于等于上堆的任意元素   
    //注意一定要和下堆堆顶比较，因为第一次插入后元素一定在下堆，如果和上堆堆顶比较就会WA 
    while(topheap.size>bottomheap.size)
        bottomheap.insert(-topheap.heap[1]),topheap.pop();
    while(bottomheap.size>topheap.size+1)
        topheap.insert(-bottomheap.heap[1]),bottomheap.pop();
    //维护上下堆平衡，保证两堆元素数相等或下堆元素数比上堆元素数多1 
} 
signed main(void){
    scanf("%d",&p);
    while(p--){
        scanf("%d%d",&a,&m);
        topheap=bottomheap=empty;
        printf("%d %d\n",a,(m+1)>>1);
        for(int i=1,x;i<=m;i++){
            scanf("%d",&x);
            add(x);
            if(i&1){
                printf("%d%c",-bottomheap.heap[1],i==m?'\n':' ');
                if(i%20==19)
                    printf("\n");
            } 
        }
    }
    return 0;
}

---

by >o< neighthorn