BZOJ1303: [CQOI2009]中位数图

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<vector>
#define MAXN 100000+10
#define a(x) a[x+MAXN]
#define b(x) b[x+MAXN]
using namespace std;
int a[MAXN<<1],b[MAXN<<1];
int n,m,p;
int s[MAXN];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&s[i]);
if(s[i]==m){
p=i;
}
}
int fe=0,bi=0;
a(0)++;
for(int i=p-1;i>=1;i--){
if(s[i]<m)fe++;
else bi++;
a(fe-bi)++;
}
fe=0,bi=0;
b(0)++;
for(int i=p+1;i<=n;i++){
if(s[i]<m)fe++;
else bi++;
b(fe-bi)++;
}
int ans=0;
for(int i=-n;i<=n;i++){
ans+=a(i)*b(-i);
}
printf("%d\n",ans);
return 0;
}