Ural 1306. Sequence Median

最新推荐文章于 2019-11-21 17:21:30 发布

原创最新推荐文章于 2019-11-21 17:21:30 发布 · 783 阅读

0 ·

CC 4.0 BY-SA版权

ural 同时被 2 个专栏收录

10 篇文章

订阅专栏

数据结构

9 篇文章

订阅专栏

本文介绍了一种利用优先队列高效求解大规模数列中位数的方法。通过将一半数量的数据压入优先队列，并随着后续数据的加入进行调整，最终找到中位数。适用于处理多达250000个整数的情况。

求一个数列的中位数，最多有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;
}