bzoj 1303: [CQOI2009]中位数图 （乱搞）

题目描述

传送门

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

题解

找到b所在的位置，对于排列中其他位置的数大于b的值为1，否则值为-1。
统计左右区间和为x的个数。然后枚举计算即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 100003
#define LL long long 
using namespace std;
LL cntl[N],cntr[N],cntl1[N],cntr1[N],ans;
int n,x,a[N];
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d%d",&n,&x); int pos=0;
    for (int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
        if(a[i]==x) pos=i;
    }
    cntl[0]=1; int sum=0; 
    for (int i=pos-1;i>=1;i--) {
        int now=(a[i]>x?1:-1);
        sum+=now;
        if (sum<0) cntl1[-sum]++;
        else cntl[sum]++;
    }
    sum=0; cntr1[0]=1;
    for (int i=pos+1;i<=n;i++) {
        int now=(a[i]>x?1:-1);
        sum+=now;
        if (sum<=0) cntr1[-sum]++;
        else cntr[sum]++;
    }
    for (int i=0;i<=n;i++) {
        ans+=cntl[i]*cntr1[i];
        ans+=cntl1[i]*cntr[i];
    }
    printf("%d\n",ans);
}