hdu 4908 BestCoder Sequence

题意：给出N和M，然后给出N个数的一个排列(permutation )，让你找出中位数为M的序列个数。

由于是 consecutive sub-sequences 所以一定和m相连，即是一个 pos-i,...,pos-1,pos,pos+1,...,pos+j 形式的序列。

只要left[大的个数]+right[大的个数]==left[小的个数]+right[小的个数]即满足条件；

设个数差x=大的个数-小的个数，那么left[x]==right[-x];统计出left和right后，答案=∑left[x]*right[-x]。

由于下标不能为负，设置n为起始，即left[n+x]==right[n-x]  x取[-n,n]。

#include<cstdio>
#include<iostream>
using namespace std;
#define mem(a,b) memset(a,b,sizeof a);
const int N=40005;
int n,m,pos,a[N],l[2*N],r[2*N];
void solve(){
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		if(a[i]==m) pos=i;
	}
	mem(l,0);mem(r,0);
	//l[0],r[0]不一定为1,所以i要以pos开始
	for(int i=pos,x=n;i>=1;i--){
		if(a[i]>m) x++;
		else if(a[i]<m) x--;
		l[x]++;
	}
	for(int i=pos,x=n;i<=n;i++){
		if(a[i]>m) x++;
		else if(a[i]<m) x--;
		r[x]++;
	}
	
	long long ans=0;
	for(int i=-n;i<=n;i++){
		ans+=1l*l[n+i]*r[n-i];
	}
	printf("%I64d\n",ans);
}
int main(){
	int tt=1,cas;
	//cin>>tt;
	for(cas=1;cas<=tt;cas++){
		while(scanf("%d %d",&n,&m)!=EOF){
			solve();
		}
	}
	return 0;
}