SPOJ.com - Problem RMID2https://www.spoj.com/problems/RMID2/题意:输入-1时,输出中位数,如果是偶数个数字的话输出小的那个;
注意的是:
priority_queue<int, vector<int>, less<int> > qmax;是大根堆,堆顶部的是最大的那个元素!!
priority_queue<int, vector<int>, greater<int> >qmin;是小根堆,顶部的是最小的那个元素!!
思路:
于是可以用一个大根堆和一个小根堆(两个优先队列)做;
大根堆可以当成 中位数及比中位数小的数字 的堆,小根堆可以当成 比中位数大的数 的堆;
所以要输出的中位数的候选项一定是在大根堆里;
也就是说,输入数字的时候
1.如果大根堆是空的,那么把数字放进去;
2. 如果输入的数字 小于等于 大根堆的堆顶的那个数,那么把它放到大根堆里去;
3.其余情况(大根堆是满的且数字大于大根堆的堆顶的数字)就放到小根堆里去;
查询的时候
1.大根堆里的数多了---->放到小根堆里去;
2.大根堆里的数小了---->从小根堆堆顶里拿数字放到大根堆里;
如何判断大根堆里的数多了少了?
两个堆里的数之和是偶数?大根堆里的个数是偶数除以2
两个堆里的数之和是奇数? 大根堆里的个数是(奇数+1)除以2
#include<bits/stdc++.h>
using namespace std;
int main()
{
int T,n;scanf("%d",&T);
while(T--){
priority_queue<int, vector<int>, less<int> > qmax;//不手动清空了,直接再建一个
priority_queue<int, vector<int>, greater<int> >qmin;//手动清空说不定会TLE
int a;
while(scanf("%d",&a)){
if(a==0)break;
else if(a==-1){//让两个堆大小平衡的代码放在这里,不然会TLE
while(qmax.size()>(qmax.size()+qmin.size()+1)/2){
qmin.push(qmax.top());
qmax.pop();
}
while(qmax.size()<(qmax.size()+qmin.size()+1)/2){
qmax.push(qmin.top());
qmin.pop();
}
printf("%d\n",qmax.top());
qmax.pop();
}
else if(a>0){
if (qmax.empty()||a<=qmax.top()) qmax.push(a);//!!!
else qmin.push(a);
}
}
}
return 0;
}