中位数问题6

Description

给出一个长度为N的非负整数序列A_i，对于所有1 < = k < = (N + 1) / 2， 输出前1,3,5,…个数的中位数。

Format

Input

第1行为一个正整数N，表示了序列长度。

N<=10^5

第2行包含N个整数A_i

(-10^9 < = A_i < = 10^9)

Output

如题所示。

Samples

输入数据 1

7
1 3 5 7 9 11 6

Copy

输出数据 1

1
3
5
6

两个优先队列解决。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<vector>
#include<math.h>
#include<iomanip>
#include<set>
#include<queue>
#include<stack> 
#include<map>
using namespace std;
int n;
priority_queue<int, vector<int>, greater<int>>qq;//返回最小
priority_queue<int>qq1;//返回最大
int main()
{
	cin >> n;
	int c,j=1;//j用来判断是前几个
	for (int i = 1; i <= n; i++)
	{
		cin >> c;
		if (qq.size() == 0 && qq1.size() == 0)
		{
			qq1.push(c);//第一个值插入小堆
		}
		else if (qq1.top() < c)
		{
			qq.push(c);//如果输入的数比小堆顶小，入小堆
		}
		else
		{
			qq1.push(c);//否则入大堆
		}
		if ((qq.size() - qq1.size()) == 2)//相差为2时转移
		{
			qq1.push(qq.top());
			qq.pop();
		}
		if ((qq1.size() - qq.size()) == 2)
		{
			qq.push(qq1.top());
			qq1.pop();
		}
		if (i == j)
		{
			if (qq.size() > qq1.size())//因为每次是奇数，所以弹出元素多的堆
				cout << qq.top()<<endl;
			else
				cout << qq1.top()<<endl;
			j = j + 2;//输出后j要加2	
		}
	}
}