AcWing.106. 动态中位数 (优先队列大小根堆)

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

输入格式

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

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

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

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

输出格式

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

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

输出中不应该存在空行。

数据范围

1≤P≤10001≤P≤1000,
1≤M≤99991≤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

 

 ~这里面所说的中位数是排完序后的中位数;

每次新加入一个数都会影响整个序列的大小顺序,中位数也就不稳定;如果每次加入一个数都用sort,然后取中间值的话,复杂度gg;所以不仅要实现动态数组,还要实现动态排序数组;可以想到优先队列的大根堆小根堆;但是呢,STL不能直接访问队列中间元素;所以就需要大小根堆结合使用;

假设序列的长度为m

  • 大根堆:存放序列前 m/2 的元素;
  • 小根堆:存放序列后 (m+1) 的元素;小根堆的堆顶就是当前序列的中位数

每次读入一个数,当大于中位数的时候就放入小根堆(后半段),小于中位数的时候就放入大根堆(前半段);

这样就实现了动态排序数组;

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int t;
    cin >>t;
    while(t--)
    {
        int id,m;
        cin >>id>>m;
        cout <<id<<" "<<(m+1)/2<<endl;
        priority_queue<int,vector<int>, greater<int> > q;//小根堆
        priority_queue<int> p;//大根堆
        int cnt=0;
        for(int i=1;i<=m;i++)
        {
            int x;
            cin >>x;
            if(q.empty()) q.push(x);
            else
            {
                if(x>q.top()) q.push(x);
                else p.push(x);
                while(q.size()>p.size()+1) p.push(q.top()),q.pop();
                while(q.size()<p.size()) q.push(p.top()),p.pop();
            }
            if(i&1)
            {
                cnt++;
                cout <<q.top()<<" ";
                if(cnt%10==0) cout <<endl;
            }
        }
        if(cnt%10) cout <<endl;
    }
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值