【Bzoj1303】中位数图

题意

给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后，位于中间的数。

---

解析

因为数都是不重复的，所以可以把左边小于它的记为+1，大于的记为-1，统计前缀和f。同理，右边大于的记为1，小于的记为-1，统计前缀和g，那么，如果左边某个数出现某次数的方法种数等于右边的，就根据乘法原理统计答案，特别地，f[0]=g[0]=1。数组下标要做一点处理。

---

#include <cstdio>
#include <algorithm>

#define Rep( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i<=(i##_END);i++)
#define For( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i++)
#define Lop( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i>=(i##_END);i--)
#define Dnt( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i--)

using std :: max;
using std :: min;

const int maxx = (100000<<1) + 25;
const int N = 100005;

int n,m,x,y,z;
int a[maxx],f[maxx],g[maxx];
int ans,tmp,pos;

int main(){
    scanf("%d%d",&n,&m);
    Rep( i , 1 , n ) scanf("%d",&a[i]);
    Rep( i , 1 , n ) if(a[i] == m) pos = i;
    f[N] = g[N] = 1;
    Dnt( i , pos-1 , 0 ) tmp += (a[i]<m)? 1 : -1,f[tmp+N] ++;
    tmp = 0;
    Rep( i , pos+1 , n ) tmp += (a[i]>m)? 1 : -1,g[tmp+N] ++;
    Rep( i , N-n , N+n ) ans += f[i]*g[i];
    printf("%d",ans);
    return 0;
}