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
}
}
}