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

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

输入格式

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

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

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

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

输出格式

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

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

输出中不应该存在空行。

数据范围

1≤P≤10001≤P≤1000,
1≤M≤999991≤M≤99999,
所有 MM 相加之和不超过 5×1055×105。

输入样例：

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 <queue>
#include <cstdio>
#include <iostream>

using namespace std;

int main()
{
    int T;
    cin >> T;
    while (T -- )
    {
        int n, m;
        scanf("%d %d", &m, &n);
        printf("%d %d\n", m, (n + 1) / 2);
        
        priority_queue<int> up;//大根堆
        priority_queue<int, vector<int>, greater<int>> down;//小根堆
        
        int cnt = 0;//题目要求一行超过10个数就换行
        for (int i = 1; i <= n; i ++ )
        {
            int x;
            scanf("%d", &x);
            
            if (up.empty() || up.top() >= x) up.push(x);
            else down.push(x);
            
            if (up.size() > down.size() + 1)
            {
                down.push(up.top());
                up.pop();
            }
            if (down.size() > up.size())
            {
                up.push(down.top());
                down.pop();
            }
            
            if (i % 2)//如果当前输入的时奇数
            {
                printf("%d ", up.top());//输出大根堆堆顶元素
                if (++ cnt % 10 == 0) printf("\n"); //若此行超过了10个数，换行
            }
        }
        
        if (cnt % 10) printf("\n");//这一轮输入结束后若不足10个数需要换行，以便下一组数据输入
    }
    
    return 0;
}