动态中位数 对顶堆

题目链接

https://www.acwing.com/problem/content/108/

依次读入一个整数序列,每当已经读入的整数个数为奇数时,输出已读入的整数构成的序列的中位数。

输入格式

第一行输入一个整数P,代表后面数据集的个数,接下来若干行输入各个数据集。

每个数据集的第一行首先输入一个代表数据集的编号的整数。

然后输入一个整数MM,代表数据集中包含数据的个数,MM一定为奇数,数据之间用空格隔开。

数据集的剩余行由数据集的数据构成,每行包含10个数据,最后一行数据量可能少于10个,数据之间用空格隔开。

输出格式

对于每个数据集,第一行输出两个整数,分别代表数据集的编号以及输出中位数的个数(应为数据个数加一的二分之一),数据之间用空格隔开。

数据集的剩余行由输出的中位数构成,每行包含10个数据,最后一行数据量可能少于10个,数据之间用空格隔开。

输出中不应该存在空行。

数据范围

1≤P≤1000
1≤M≤9999

输入样例:

3 
1 9 
1 2 3 4 5 6 7 8 9 
2 9 
9 8 7 6 5 4 3 2 1 
3 23 
23 41 13 22 -3 24 -31 -11 -8 -7 
3 5 103 211 -311 -45 -67 -73 -81 -99 
-33 24 56

输出样例:

1 5
1 2 3 4 5
2 5
9 8 7 6 5
3 12
23 23 22 22 13 3 5 5 3 -3 
-7 -3

 题解

        为了动态维护中位数,我们可以建立两个二叉堆:一个小根堆,一个大根堆。再依次读入这个整数序列的过程中,设当前序列长度为M,我们始终保持:

1.序列从小到大排名为1到M/2的整数保存在大根堆里;

2.序列从小到大排名在M/2+1到M的整数保存在小根堆里

   任何时候,如果某一个堆中元素过多,打破这个性质,就取出该堆的堆顶插入另一个堆。这样一来,序列的中位数就是小根堆的堆顶。

   每次读入一个数值X后,或X比中位数小,则插入到大根堆中,否则插入小根堆中,在插入之后检查并维护上述性质即可。这就是“”对顶堆”算法。

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
int main(){
   int t;
   cin>>t;
   while(t--){
    priority_queue<int> maxx;
    priority_queue<int,vector<int>,greater<int> > minn;
    int x,y;
    cin>>x>>y;
    cout<<x<<" "<<(y+1)/2<<endl;
    int cnt=0;
    for(int i=0;i<y;i++){
        int a;
        cin>>a;
        if(minn.empty()||minn.top()<a){
             minn.push(a);
             if(minn.size()>maxx.size()+1){
                int b=minn.top();
                minn.pop();
                maxx.push(b);
             }
        }else{
            maxx.push(a);
            if(maxx.size()>minn.size()){
                int b=maxx.top();
                maxx.pop();
                minn.push(b);
            }
        }
        if(i%2==0){
            cout<<minn.top()<<" ";
            cnt++;
        }
        if(cnt==10){
            cnt=0;
            cout<<endl;
        }
    }
    if(cnt!=0) cout<<endl;
   }
    return 0;
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值