动态中位数（对顶堆）

动态中位数

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

输入格式

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

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

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

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

输出格式

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

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

输出中不应该存在空行。

数据范围

1≤P≤1000,
1≤M≤99999,
所有 M 相加之和不超过 5×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

算法：对顶堆

朴素做法，当数据较强时可能TLE

#include <iostream>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
    int n;
    cin >> n;
    while(n--)
    {
        int k, w;
        cin >> k >> w;
        vector<int> nums;
        priority_queue<int, vector<int>, greater<int>> min_heap;
        priority_queue<int> max_heap;
        for(int i = 1; i <= w; i++)
        {
            int num;
            cin >> num;
            if(i & 1) 
            {
                min_heap.push(num);
                if(!max_heap.empty() && min_heap.top() < max_heap.top())
                {
                    int x = min_heap.top(); min_heap.pop();
                    int y = max_heap.top(); max_heap.pop();
                    max_heap.push(x), min_heap.push(y);
                }
                nums.push_back(min_heap.top());
            }
            else
            {
                min_heap.push(num);
                max_heap.push(min_heap.top());
                min_heap.pop();
            }
        }
        cout << k << ' ' << nums.size() << endl;
        for(int i = 0; i < nums.size(); i++)
        {
            cout << nums[i] << ' ';
            if((i + 1) % 10 == 0 && nums.size() > i + 1) puts("");
        }
        puts("");
    }
    return 0;
}

优化

我用的vector存储所有中位数，其实可以直接输出来进一步优化

#include <iostream>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
    int n;
    cin >> n;
    while(n--)
    {
        int k, w;
        cin >> k >> w;
        vector<int> nums;
        priority_queue<int, vector<int>, greater<int>> min_heap;
        priority_queue<int> max_heap;
        for(int i = 1; i <= w; i++)
        {
            int num;
            cin >> num;
            if(!max_heap.empty() && num < max_heap.top()) max_heap.push(num);
            else min_heap.push(num);
            if(min_heap.size() > max_heap.size()) 
            {
                max_heap.push(min_heap.top());
                min_heap.pop();
            }
            if(max_heap.size() == min_heap.size() + 2)
            {
                min_heap.push(max_heap.top());
                max_heap.pop();
            }
            if(i & 1) nums.push_back(max_heap.top());
        }
        cout << k << ' ' << nums.size() << endl;
        for(int i = 0; i < nums.size(); i++)
        {
            cout << nums[i] << ' ';
            if((i + 1) % 10 == 0 && nums.size() > i + 1) puts("");
        }
        puts("");
    }
    return 0;
}