求以m为中位数的连续子序列的个数。
那么统计m位置往右的大于(小于)m的个数 相等 位置往左小于(大于)m的个数 的个数即可。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[40005],b[80005];
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
int pos;
memset(b,0,sizeof(b));
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
if(a[i]==m) pos=i;
}
int cnt=0,ans=0;
b[n]=1;
for(int i=pos+1;i<n;i++)
{
if(a[i]>m) cnt++;
else cnt--;
b[n+cnt]++;
}
cnt=0;
for(int i=pos-1;i>=0;i--)
{
if(a[i]>m) cnt++;
else cnt--;
ans+=b[n-cnt];
}
ans+=b[n];
printf("%d\n",ans);
}
}