《算法竞赛进阶指南》动态中位数

动态中位数

动态中位数
题目
提交记录
讨论
题解
视频讲解

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

输入格式
第一行输入一个整数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

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
#pragma G++ optimize (2)
#pragma C++ optimize (2)
const int maxn=10000+10;
int main(){
        ios::sync_with_stdio(false);
        int T;
        cin>>T;
        while(T--)
        {
                priority_queue<int > max_heap;
                priority_queue<int ,vector<int  > ,greater<int > >min_heap;//通过建立大根堆和小根堆
                //大根堆存放数据的小的一部分，小根堆存放数据大的一部分，这样大根堆对顶就是动态存储中位数
                //大根堆是降序的，小根堆是升序的
                int n,m,cnt=0;
                cin>>n>>m;
                cout<<n<<" "<<m/2+1<<endl;
                for(int i=0;i<m;i++)
                {
                        int k;
                        cin>>k;
                        max_heap.push(k);//首先将数据存储到大根堆的堆顶，然后在比较是否交换两堆的堆顶
                        if(min_heap.size()&&max_heap.top()>min_heap.top())
                        {    //max_heap存放小数据 min_heap存放大数据
                                auto a=max_heap.top(),b=min_heap.top();
                                max_heap.pop();
                                min_heap.pop();
                                max_heap.push(b);
                                min_heap.push(a);
                        }
                        if(max_heap.size()>min_heap.size()+1)
                        {  //大根堆和小根堆应均分数据
                                min_heap.push(max_heap.top());
                                max_heap.pop();
                        }
                        if(!(i&1))
                        {
                                cout<<max_heap.top()<<" ";
                                if((++cnt)%10==0)
                                cout<<endl;
                        }
                }
                if(!cnt%10==0)cout<<endl;
        }
        return 0;
}