传送门

把中位数出现的位置设为 $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);
}