P1168 中位数

题目描述

给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[3], …, 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[3], …, 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。

思路：维护两个堆，一个大根堆存放<=中位数的值，小根堆存放>=中位数的值，保证小根堆比大根堆规模大1。

维护的时候可以先根据上次mid暴力插入，再调整规模。复杂度显然是O(nlogn)的。

#include<cstdio>
#include<iostream>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;

int n,a,b,m;
priority_queue<int> L;
priority_queue<int,vector<int>,greater<int> > R;

int main()
{
	scanf("%d%d",&n,&m);
	printf("%d\n",m);
	L.push(m);
	
	for(int i=1;i<=(n-1)/2;i++)
	{
		scanf("%d%d",&a,&b);
		if(a>=m)R.push(a); else L.push(a);
		if(b>=m)R.push(b); else L.push(b);
		while(L.size()>=R.size()){R.push(L.top());L.pop();}
		while(R.size()>L.size()+1){L.push(R.top());R.pop();}
		m=R.top();
		printf("%d\n",m);
	}
	
	return 0;
}