求一个数列的中位数,最多有250000个数。
看了discuss才知道用优先队列,先压栈N/2+1个然后边压栈边弹栈。
N为奇数就是栈里的弹一个,N为偶数就从栈里弹两个。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
priority_queue<unsigned int> que;
int main()
{
//freopen("in","r",stdin);
int n;
unsigned int u;
unsigned int ans;
scanf("%d",&n);
for(int i=0;i<=n/2;i++)
{
scanf("%u",&u);
que.push(u);
}
for(int i=n/2+1;i<n;i++)
{
scanf("%u",&u);
que.push(u);
que.pop();
}
if(n%2)
{
ans=que.top();
printf("%u\n",ans);
}
else
{
ans=que.top();
que.pop();
unsigned int tem=que.top();
printf("%u",(tem+ans)/2);
if((tem+ans)%2)
printf(".5");
printf("\n");
}
return 0;
}