题目描述
给出一个长度为N的非负整数序列A[i],对于所有1 ≤ k ≤ (N + 1) / 2,输出A[1], A[3], …, A[2k - 1]的中位数。即前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]的中位数。
输入样例
7
1 3 5 7 9 11 6
输出样例
1
3
5
6
说明
对于20%的数据,N ≤ 100;
对于40%的数据,N ≤ 3000;
对于100%的数据,N ≤ 100000.
题解:
建一个小根堆,一个大根堆,以根为mid值,大数扔进小根堆,小数扔进大根堆,保证两个堆元素总量相等即可,根就是题目所求中位数
code
#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int a[100005];
priority_queue<int>q1;
priority_queue<int,vector<int>,greater<int> >q2;
int main()
{
int n,mid,i;
scanf("%d",&n);
scanf("%d",&a[1]);
mid=a[1];
printf("%d\n",mid);
for(i=2;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]>mid)
q2.push(a[i]);//比中间数大,入小根堆
else
q1.push(a[i]);//否则,入大根堆
if(i%2)//奇数,调整两个堆的大小
{
while(q1.size()!=q2.size())
{
if(q1.size()>q2.size())
{
q2.push(mid);
mid=q1.top();
q1.pop();
}
else
{
q1.push(mid);
mid=q2.top();
q2.pop();
}
}
printf("%d\n",mid);
}
}
return 0;
}