洛谷 P1168 中位数

题目描述

给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[2], …, A[2k - 1]的中位数。[color=red]即[/color]前1，3，5，……个数的中位数。

输入输出格式

输入格式：

输入文件median.in的第1行为一个正整数N，表示了序列长度。

第2行包含N个非负整数A[i] (A[i] ≤ 10^9)。

输出格式：

输出文件median.out包含(N + 1) / 2行，第i行为A[1], A[2], …, A[2i – 1]的中位数。

输入输出样例

输入样例#1：

7
1 3 5 7 9 11 6

输出样例#1：

1
3
5
6

说明

对于20%的数据，N ≤ 100；

对于40%的数据，N ≤ 3000；

对于100%的数据，N ≤ 100000。

将比中位数大的放到一个小根堆，剩下的放到一个大根堆，每次调整大小。

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
int n;
priority_queue<int,vector<int>,greater<int> >sm;
priority_queue<int>bg;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		if(sm.empty()||x>sm.top())
			sm.push(x);
		else
			bg.push(x);
		if(i%2==1)
		{
			while(sm.size()>bg.size()+1)
			{
				bg.push(sm.top());
				sm.pop();
			}
			while(sm.size()<bg.size()+1)
			{
				sm.push(bg.top());
				bg.pop();
			}
			printf("%d\n",sm.top());
		}
	}
	return 0;
}