百度之星 hdu5701 中位数计数

http://acm.hdu.edu.cn/showproblem.php?pid=5701

给出一个序列，取其中的任何一个连续的序列，该序列的数从小到大排列，待更新，，，

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN = 8005;
int num[MAXN];
int arr[MAXN];
int sum[MAXN*2];
int cnt[MAXN];
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i = 1;i<=n;i++)
            scanf("%d",&num[i]);
        for(int i = 1;i<=n;i++)
        {
            int temp = num[i];
            memset(arr,0,sizeof(arr));
            memset(sum,0,sizeof(sum));
            int res = 0;
            for(int j = i-1;j>=1;j--)
            {
                arr[j] = arr[j+1]+((num[j]>temp)?-1:1);
                sum[arr[j]+n]++;
                if(arr[j]==0)
                    res++;
            }
            for(int j = i+1;j<=n;j++)
            {
                arr[j] = arr[j-1]+((num[j]>temp)?1:-1);
                if(sum[arr[j]+n]>0)
                    res=res+sum[arr[j]+n];
                if(arr[j]==0)
                    res++;
            }
            cnt[i] = res+1;
        }
        printf("%d",cnt[1]);
        for(int i = 2;i<=n;i++)
            printf(" %d",cnt[i]);
        printf("
");
    }
    return 0;
}