【BZOJ 1303】 [CQOI2009]中位数图

【链接】 我是链接,点我呀:)
【题意】

【题解】

往左走求个前缀和(1和-1)now
然后答案累加dic[-now]就可以了

【代码】

#include <bits/stdc++.h>
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define all(x) x.begin(),x.end()
#define pb push_back
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
using namespace std;

const double pi = acos(-1);
const int dx[4] = {0,0,1,-1};
const int dy[4] = {1,-1,0,0};
const int N = 1e5;

int n,r,idx = 1;
int a[N+10];
map<int,int> dic;

int main(){
	#ifdef LOCAL_DEFINE
	    freopen("rush_in.txt", "r", stdin);
	#endif
    scanf("%d%d",&n,&r);
    rep1(i,1,n) scanf("%d",&a[i]);
    rep1(i,2,n) if (a[i]==r) idx = i;
    LL now = 0;
    dic[0]++;
    rep1(i,idx+1,n) {
        now += (a[i]>r?1:(-1));
        dic[now]++;
    }
    now = 0;
    LL ans = 0;
    rep2(i,idx-1,1){
        now+=(a[i]>r?1:(-1));
        ans+=dic[-now];
    }
    ans+=dic[0];
    printf("%lld\n",ans);
	return 0;
}