[CQOI2009]中位数

传送门

把中位数出现的位置设为$p$，比中位数大的设为$1$，小的设为$-1$。$s[i]$表示$i$到$p$的和，$l[s[i]]$表示$p$左边和为$s$出现的次数，$r[s[i]]$同理，如果$l[i]=r[-i]$则$ans+=l[i]\times r[-i]$.

#include<cstdio>
int b,n,p,x,ans,a[400005],s[200005],l[400005],r[400005];
int main(){
	scanf("%d%d",&n,&b);
	for(int i=1;i<=n;++i) scanf("%d",&x),x==b?p=i,a[i]=0:a[i]=x>b?1:-1;
	l[n]=r[n]=1;
	for(int i=p-1;i>=1;--i) s[i]=s[i+1]+a[i],++l[s[i]+n];
	for(int i=p+1;i<=n;++i) s[i]=s[i-1]+a[i],++r[s[i]+n];
	for(int i=0;i<2*n;++i) ans+=l[i]*r[n-i+n];
	return !printf("%d",ans);
}