hdu5701中位数计数(百度之星)

来源:互联网 发布:linux fm驱动 编辑:程序博客网 时间:2024/05/18 01:41

中位数计数

Time Limit: 12000/6000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1734 Accepted Submission(s): 631

Problem Description
中位数定义为所有值从小到大排序后排在正中间的那个数,如果值有偶数个,通常取最中间的两个数值的平均数作为中位数。

现在有n个数,每个数都是独一无二的,求出每个数在多少个包含其的区间中是中位数。

Input
多组测试数据

第一行一个数n(n≤8000)

第二行n个数,0≤每个数≤109,

Output
N个数,依次表示第i个数在多少包含其的区间中是中位数。

Sample Input
5
1 2 3 4 5

Sample Output
1 2 3 2 1

向一个方向进行统计,每出现一个比当前数小的数,用num–记录,出现比他大的数则
num++记录,若num为0,则证明发现一个区间(一个比它大,一个小,刚好位于中间数),若每边的查询后发现有不对称的(比如3 5 6 4 10 8 7 以4向左查询,则num为1,我们用vist记录vist[num]++,再对右边进行处理时则有num为1,2,3,分别标记vist[1]++,vist[2]++,vist[3]++,这样在右边统计区间时,由于一边一大一小刚好形成一区间,要多加上vist[-num])
遍历每个数,分别向左和向右统计其区间数量,时间复杂度为O(N^2),8000*8000=6.4*1e7,时间足够

#include<iostream>#include<algorithm>#include <cstdio>#include <cstring>using namespace std;int main(){    int n;    int a[8010];    int vist[2*8010];    long long ans[8010];    while(scanf("%d",&n)!=EOF)    {        for(int i=0; i<n; i++)            scanf("%d",&a[i]);        //int num=0;        memset(ans,0,sizeof(ans));        for(int i=0; i<n; i++)        {            int num=0;            memset(vist,0,sizeof(vist));            for(int j=i-1; j>=0; j--)            {                if(a[j]<a[i])num--;                else num++;                if(num==0) ans[i]++;                vist[num+8010]++;            }            num=0;            for(int j=i+1; j<n; j++)            {                if(a[j]<a[i])num--;                else num++;                if(num==0) ans[i]++;                ans[i]+=vist[-num+8010];            }        }        for(int i=0; i<n-1; i++)        {            printf("%I64d ",ans[i]+1);        }        printf("%I64d\n",ans[n-1]+1);    }    return 0;}