题意:给出n个数,求出前i个数的中位数(i<n并且i是奇数)。
分析:对顶堆板子题,也可以用链表+hash做,不过比较复杂,详见《算法竞赛进阶指南》P36。
代码:
#include<stdio.h>
#include<queue>
using namespace std;
int t,n,m,u,mid,cas;
/*
用最大堆维护比当前中位数小的,用最小堆
维护比当前中位数大的,如果两个堆大小差
为2,则将中位数加入大小较小的堆,并且
将另一个堆的顶作为新的中位数
*/
int main(){
scanf("%d",&t);
while(t--){
priority_queue<int> qu;
priority_queue<int,vector<int>,greater<int> >qu1;
scanf("%d%d",&n,&m);
printf("%d %d\n",n,(m+1)/2);
for(int i=1;i<=m;i++){
scanf("%d",&u);
if(i==1){
cas=1,mid=u;
printf("%d ",mid);
continue;
}
if(u<mid) qu.push(u);
else qu1.push(u);
if(qu.size()-qu1.size()==2){
qu1.push(mid);
mid=qu.top();
qu.pop();
}
else if(qu1.size()-qu.size()==2){
qu.push(mid);
mid=qu1.top();
qu1.pop();
}
if(i%2!=0){
cas++;
if(cas%10==0) printf("%d\n",mid);
else printf("%d ",mid);
}
}
if(cas%10!=0) printf("\n");
}
return 0;
}