POJ3784 Running Median 对顶堆

题目链接

http://poj.org/problem?id=3784

分析

动态维护序列中位数可用对顶堆实现，即用大根堆维护序列前一段，用小根堆维护序列后一段。

保证大根堆内元素均比小根堆内元素小，且两堆元素个数差不超过 $1$。

每次输出时，元素个数较多的堆的堆顶元素即为当前序列中位数。

AC代码

#include <cstdio>
#include <vector>
#include <queue>

using namespace std;

inline int read() {
	int num = 0, flag = 1;
	char c = getchar();
	while (c < '0' || c > '9') {
		if (c == '-') flag = -1;
		c = getchar();
	}
	while (c >= '0' && c <= '9')
		num = num * 10 + c - '0', c = getchar();
	return flag * num;
}

priority_queue<int> q1;
priority_queue<int, vector<int>, greater<int> > q2;

int main() {
	int t = read();
	while (t--) {
		while (!q1.empty()) q1.pop();
		while (!q2.empty()) q2.pop();
		int n = read(), m = read();
		printf("%d %d\n", n, m / 2 + 1);
		for (int i = 1; i <= m; ++i) {
			int x = read();
			if (q1.empty() || x <= q1.top()) q1.push(x);
			else q2.push(x);
			while (q1.size() > q2.size() + 1) q2.push(q1.top()), q1.pop();
			while (q2.size() > q1.size() + 1) q1.push(q2.top()), q2.pop();
			if (i > 1 && i % 20 == 1) puts("");
			if (i & 1)
				printf("%d ", q1.size() > q2.size() ? q1.top() : q2.top());
		}
		if (t) puts("");
	}
	return 0;
}