【优先队列】ACW 106. 动态中位数

题目

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

输入格式

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

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

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

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

输出格式

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

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

输出中不应该存在空行。

数据范围

1≤P≤1000,
1≤M≤9999

解析

本题在leetcode也有类似的题目，本质上在于使用一个最大堆和一个最小堆来维护中间元素，具体可参照代码注释

代码

#include <iostream>
#include <cstdlib>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
#include <queue>
#include <functional>
#include <cstdio>

using namespace std;

int main() {
    int T;
    cin >> T;
    while (T--) {
        int n, m;
        cin >> m >> n;
        cout << m << " " << int((n + 1) / 2) << endl;

        // 最大堆，从大到小排列，放置比中位数要小的元素
        priority_queue<int, vector<int>, less<>> down;
        // 最小堆，从小到大排列，放置比中位数要大的元素
        priority_queue<int, vector<int>, greater<>> up;

        int cnt = 0;
        for (int i = 1; i <= n; i++) {
            int x;
            cin >> x;
            
            // 加入的元素无非就两种情况，加入down或up
            // 以down为中位数出口的话优先加入down
            if(down.empty() || x <= down.top()) down.push(x);
            else up.push(x);
            
            // 加入后动态调整堆的元素个数
            if(down.size() > up.size() + 1) {
                up.push(down.top());
                down.pop();
            }
            if(up.size() > down.size()){
                down.push(up.top());
                up.pop();
            }

            if (i % 2) {
                cout << down.top() << " ";
                if (++cnt % 10 == 0) puts("");
            }
        }

        if (cnt % 10) puts("");
    }

    return 0;
}