51Nod - 1682 ：中位数计数

中位数计数

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

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

Input 第一行一个数n(n<=8000)
第二行n个数，0<=每个数<=10^9 Output N个数，依次表示第i个数在多少包含其的区间中是中位数。 Sample Input

5
1 2 3 4 5

Sample Output

1 2 3 2 1

a[i]是区间的中位数 即 区间内 大于a[i]的数的个数A = 小于a[i] 的数的个数B；

即 i 左边的 A-B 的值 = i  右边的 B - A 的值；

cnt 为 x-i 区间内 A - B 的值 （x∈[ 0 , i )）；

对于每一个 a[i]，在 0 - i 求一个 m 数组；

m[cnt] 表示 [ 0 , i ) 内 A - B = cnt 的区间终点为i的区间起点个数；

初始化 m[0] =1，此区间只包含 a[i] ;

ans[i] 为 i 是中位数 区间的个数

遍历[ i ,n ) 再算一遍 cnt 则此时的区间个数即 m[-1*cnt] ;

（我写的 cnt 为 B - A ,则此时的区间个数为 m[cnt]）

最后 ans[i] 求一遍和即可；

#include <iostream>
#include <cstdio>
#include <cstring>
#define ll long long
using namespace std;
ll a[8050];int ans[8050];
int m[16050];
int main(){
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
        scanf("%lld",&a[i]);
    int cnt;
    for(int i=0;i<n;i++){
        memset(m,0,sizeof(m));
        m[8000]=1;
        ans[i]=0;
        cnt=0;
        for(int j=i-1;j>=0;j--){
            if(a[j]>a[i]) cnt++;
            else cnt--;
            m[cnt+8000]++;
        }
        cnt=0;
        for(int j=i;j<n;j++){
            if(a[j]>a[i]) cnt--;
            else if(a[j]<a[i]) cnt++;
            ans[i]+=m[cnt+8000];
        }
    }
    for(int i=0;i<n;i++){
        if(i) printf(" ");
        printf("%d",ans[i]);
    }
    cout<<endl;
    return 0;
}