acwing106

如果说对静态的数据找中位数,那么用快排或直接用partion方法来做即可。但当前题目要求的是动态的读取数据来输出中位数,即输出前1、3、5、7...个数据的中位数,那么就需要使用对顶堆来处理。
对顶堆使用两个堆,上面的堆up是小根堆,下面的堆down是大根堆,且必须满足小根堆的top要大于大根堆的top。每读入一个新的数据i,先判断数据i与大根堆top值mid的大小关系,如果i更大,就将其加入小根堆up,否则加入大根堆down。然后是关键的平衡步骤:根据当前已处理的数据个数,进行操作:如果两个堆中共有奇数个元素,则down必须比up大一个元素,通过top与pop操作实现;如果共有偶数个元素,则up和down的元素个数必须相同,通过top与pop操作实现。经过这样初始化和操作后的堆在输入奇数个元素时,中位数为down的top;在输入偶数个元素时,中位数为down的top和up的top。
C++的STL已有优先队列容器,其内部实现正是堆,只需要指定其是小根堆还是大根堆即可使用。
代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<string>
using namespace std;

int P, id,M;

int main(void) {
	//加速IO
	ios::sync_with_stdio(false);

	cin >> P;
	int c = 0;
	while (++c<=P) {
		priority_queue<int, vector<int>, greater<int>> up;	//小根堆
		priority_queue<int, vector<int>, less<int>> down;	//大根堆
		cin >>id>> M;

		cout << id << " " << (M + 1) / 2 << endl;

		int cnt = 0;	//记录已经输出的结果数
		for (int i = 1; i <= M; i++) {
			int x;	cin >> x;
			if (down.size() == 0)	down.push(x);
			else {
				if (x > down.top())
					up.push(x);
				else
					down.push(x);
			}

			if (i % 2 == 0) {
				//平衡两个堆的大小
				if (down.size() != up.size()) {
					up.push(down.top());
					down.pop();
				}
			}
			else {
				if (down.size() < up.size()) {
					down.push(up.top());
					up.pop();
				}
				
				//输出当前中位数
				cout << down.top();
				cnt++;
				if (cnt % 10 == 0 || cnt==(M+1)/2)
					cout << endl;
				else
					cout << " ";
			}
		}
	}
	return 0;
}
posted @ 2022-06-01 19:32  带带绝缘体  阅读(19)  评论(0编辑  收藏  举报