求解中位数

求解中位数的方法:

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;

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值