hdu 5701 中位数计数 思路题

中位数计数

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

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

 

Source

2016"百度之星" - 初赛（Astar Round2B）

思路：枚举每一位为中位数找个数；

　　　将大于这个数设为1，否则为-1；前后遍历找大于它或小于，具体看代码；

#include<bits/stdc++.h>
using namespace std;
#define ll __int64
#define mod 1000000007
#define inf 999999999
#define pi 4*atan(1)
//#pragma comment(linker, "/STACK:102400000,102400000")
const int N=1e5+10,M=2*1e5+10;
int a[N],b[N];
ll ans[N];
int l[M];
int r[M];
int main()
{
    int x,y,z,i,t;
    int T;
    while(~scanf("%d",&x))
    {
        memset(ans,0,sizeof(ans));
        for(i=1;i<=x;i++)
        scanf("%d",&a[i]);
        for(i=1;i<=x;i++)
        {
            for(t=1;t<=x;t++)
            {
                if(a[t]==a[i])
                b[t]=0;
                else
                b[t]=a[t]>a[i]?1:-1;
            }
            int flag=8000;
            for(t=i;t>0;t--)
            flag+=b[t],l[flag]++;
            flag=8000;
            for(t=i;t<=x;t++)
            flag-=b[t],r[flag]++;
            for(t=0;t<=16010;t++)
            ans[i]+=(ll)l[t]*r[t],l[t]=r[t]=0;
        }
        for(i=1;i<=x;i++)
        printf("%I64d%c",ans[i],(i==x)?'
':' ');
    }

    return 0;
}