[BZOJ 1303] CQOI 2009 中位数图 · 数学

一个大于b的数和一个小于b的数可以互相抵消，所以我们用1和-1表示。

从b向两边扩展，left[i]表示b左边抵消后有i个数比b小的可能数，right[i]表示b右边抵消后有i个数比b大的可能数。

乘法原理得ans=sigma(left[i]*right[i]).

#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <iostream>
using namespace std;

const int N=100005;
int n,m,ad,data[N*2],r[N*2],l[N*2],sum,ans;

int main(){
	cin>>n>>m;
	for (int i=1;i<=n;i++){
		cin>>data[i];
		if (data[i]==m) ad=i;
	}
	
	sum=0;l[N]=r[N]=1;
	for (int i=ad-1;i;i--)
		sum+=(data[i]<m)?1:-1,
		l[sum+N]++;
	sum=0;
	for (int i=ad+1;i<=n;i++)
		sum+=(data[i]>m)?1:-1,
		r[sum+N]++;
		
	for (int i=N-n;i<=N+n;i++)
		 ans+=r[i]*l[i];
	cout<<ans<<endl;
	return 0;
}