bzoj 1303: [CQOI2009]中位数图

#include<cstdio>
#include<iostream>
#define M 100005
using namespace std;
int a[M],l[2*M],r[2*M],sum[M],n,m,p,ans;
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
      {
        scanf("%d",&a[i]);
        if(a[i]<m)
          a[i]=-1;
        if(a[i]==m)
          {
            a[i]=0;
            p=i;
          }
        if(a[i]>m)
          a[i]=1;
      }
    l[n]=r[n]=1;
    for(int i=p-1;i;i--)
      {
        sum[i]=sum[i+1]+a[i];
        l[n+sum[i]]++;
      }
    for(int i=p+1;i<=n;i++)
      {
        sum[i]=sum[i-1]+a[i];
        r[n+sum[i]]++;
      }
    for(int i=0;i<=2*n;i++)
      ans+=l[i]*r[2*n-i];
    printf("%d
",ans);
    return 0;
}

分别统计出在中位数左边右边比他大X（比他小X）的数有多少种方案，比他小赋值为-1，比他大赋值为1，用前缀和。最后相应的相乘。