中位数图

#include<bits/stdc++.h>
//#define int long long
const int N=1e8+10;
using namespace std;
int n,x,y,a[N],b,ans=0,l,m,k;
int s1[N],s2[N];
map<int,int>mp;
signed main()
{
    cin>>n>>b;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        if(a[i]<b)a[i]=-1;//小于b的赋为-1
        if(a[i]>b)a[i]=1;//大于b的赋为1
        if(a[i]==b)
        {
            a[i]=0;//等于b时，记录b的位置，赋a[i]为0
            k=i;
        }
    }
    for(int i=k-1;i>=1;i--)
    {
        s1[i]=s1[i+1]+a[i];//左边前缀和
        mp[s1[i]]++;//记录左边出现前缀和不同值的次数
        if(s1[i]==0)ans++;
    }
    for(int i=k+1;i<=n;i++)
    {
        s2[i]=s2[i-1]+a[i];
        ans+=mp[-s2[i]];//如果左边前缀和等于右边前缀和的相反数，即相加可为0，则增加计数
        if(s2[i]==0)ans++;
    }
    cout<<ans+1;
    return 0;
}