Running Median Again——对顶堆

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;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值