51nod-1682 中位数计数

记录一个菜逼的成长。。

题目链接

考虑如何计算一个数的答案：
定义一个b数组，把所有大于自己的值定义为1，小于自己的值定义为-1，等于自己的定义为0。
如果包含这个数的区间这个数是中位数，则定义后的总和为0。
考虑前缀和，那么问题就变成了在这个数两边有多少对相等的数。
可以先把左边的数统计一下，右边的数就可以$O(1)$查询，时间复杂度$O(n)$
总时间复杂度$O(N^2)$。

这里注意一下$sum[i-1] == sum[j]$ 表示的区间才是i-j.
然后这题卡了map，只能用数组计数，又因为有负数所以把所有的前缀和加上8000再统计

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define cl(a,b) memset(a,b,sizeof(a))
const int maxn = 8000 + 10;
int a[maxn],sum[maxn],b[maxn];
int cnt[maxn*2];
vector<int>ve;
int main()
{
    int n;
    while(~scanf("%d",&n)){
        for( int i = 1; i <= n; i++ ){
            scanf("%d",a+i);
        }
        int ans;
        for( int i = 1; i <= n; i++ ){
            sum[0] = 0;
            ans = 0;
            cl(cnt,0);
            cnt[8000]++;
            for( int j = 1; j <= n; j++ ){
                if(a[j] < a[i])b[j] = -1;
                if(a[j] == a[i])b[j] = 0;
                if(a[j] > a[i])b[j] = 1;
                sum[j] = sum[j-1] + b[j];
                if(j < i)cnt[sum[j]+8000] ++;
                if(j >= i)ans += cnt[sum[j]+8000];
            }
            ve.pb(ans);
        }
        for( int i = 0; i < ve.size(); i++ ){
            printf("%d ",ve[i]);
        }
    }
    return 0;
}