动态中位数——堆

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

输入格式
第一行输入一个整数 P,代表后面数据集的个数,接下来若干行输入各个数据集。
每个数据集的第一行首先输入一个代表数据集的编号的整数。
然后输入一个整数 M,代表数据集中包含数据的个数,M 一定为奇数,数据之间用空格隔开。
数据集的剩余行由数据集的数据构成,每行包含 10 个数据,最后一行数据量可能少于 10 个,数据之间用空格隔开。

输出格式
对于每个数据集,第一行输出两个整数,分别代表数据集的编号以及输出中位数的个数(应为数据个数加一的二分之一),数据之间用空格隔开。
数据集的剩余行由输出的中位数构成,每行包含 10 个数据,最后一行数据量可能少于 10 个,数据之间用空格隔开。
输出中不应该存在空行。

数据范围
1≤P≤1000,1≤M≤99999,所有 M 相加之和不超过 5×105。

输入样例:

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 
using namespace std;
#define int long long 
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
typedef pair PII;
const int N=2e6+10;
int t,n,m;
signed main()
{
    ios;
    cin>>t;
    while (t--)
    {
        cin>>n>>m;
        cout< p;  //大根堆
        priority_queue ,greater> q;  //小根堆
        int cnt=1;
        for (int i=1;i<=m;i++)
        {
            int x;
            cin>>x;
            q.push(x);
            if (p.size()&&p.top()>q.top())
            {
                q.push(p.top());
                p.pop();
                p.push(q.top());
                q.pop();
            }
            if (q.size()>p.size()+1)
            {
                p.push(q.top());
                q.pop();
            }
            if (i&1) 
            {
                cout<

你可能感兴趣的:(算法,数据结构)