求解中位数的方法:
1.排序+取index=n/2位置的元素,每次查询代价是O(nlogn)。
2.利用堆:
中位数本身的性质是,在这串数中,一半的数比他小,一半的数比他大,那么我们可以设置一个大根堆用来存比较小的n/2个数,用一个小根堆存较大的n/2个数。
维护方法:代价:O(logn)
1.初始化的时候,将第一个数放入大根堆
2.开始读下一个数,如果这个数比当前大根堆的元素小,放入大根堆,否则进入小根堆
3.判断大根堆和小根堆,如果两个堆中元素数量差值超过1【保持两个堆大小差值在1以内】,进行如下操作:
A.如果小根堆元素多,则将小根堆堆顶弹出放入大根堆
B.如果大根堆元素多,则将大根堆堆顶弹出放入小根堆
查询方法:代价:O(1)
1.如果大根堆和小根堆元素个数不同:
A.大根堆元素比小根堆多1,返回大根堆堆顶元素
B.小根堆元素比大根堆多1,返回小根堆堆顶元素
2.如果大根堆和小根堆元素个数相同:
中位数=大根堆堆顶元素和小根堆堆顶元素的平均值
#include<cstdio>
#include<queue>
using namespace std;
priority_queue<int>big;//默认的是大根堆
priority_queue<int,vector<int>,greater<int> > small;//小根堆
int main(){
int n,r,op,bigsize,smallsize;
float result;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d%d",&op,&r); //op为0表示单纯插入 op为1表示插入后查询中位数
if(i==1){
big.push(r);
}else{
big.top()>r?big.push(r):small.push(r);
bigsize=big.size();
smallsize=small.size();
if(bigsize-smallsize>1){
r=big.top();
big.pop();
small.push(r);
}
else if(smallsize-bigsize>1){
r=small.top();
small.pop();
big.push(r);
}
}
if(op){
result=big.size()>small.size()?big.top():small.top();
result=(big.size()==small.size()?(big.top()+result)/2:result);
printf("当前中位数%.2f 当前小根堆堆大小%d 当前大根堆大小:%d\n",result,small.size(),big.size());
}
}
return 0;
}