将大于M的数标记为1 小于M的数标记为-1 M本身标记为0
则题目就是要求和为0并且包括M的连续序列的个数
用sum[i]表示从第1个数大牌第i个数的标记的和
对于所有大于等于M的位置的i 我们要求小于M的位置的sum[j]==sum[i]的个数的和即为答案
#include<bits.stdc++.h>
using namespace std;
int sum[40020],f[80020];
int main()
{
int N,M,k,a;
while(scanf("%d%d",&N,&M)==2)
{
memset(f,0,sizeof(f));
memset(sum,0,sizeof(sum));
for(int i=1;i<=N;i++)
{
scanf("%d",&a);
if(a>M)
sum[i]=sum[i-1]+1;
else if(a<M)
sum[i]=sum[i-1]-1;
else
{
k=i;
sum[i]=sum[i-1];
}
}
for(int i=0;i<=k-1;i++)
f[sum[i]+40000]++;
int ans=0;
for(int i=k;i<=N;i++)
ans+=f[sum[i]+40000];
printf("%d\n",ans);
}
return 0;
}