bzoj1303: [CQOI2009]中位数图

题目传送门
好题。

解法：
连续子段中位数为b。
那么这段子段肯定包含b。
那么从b的位置开始可以往左边延伸也可以往右边延伸。
那么小于b的数的数量要等于大于b的数的数量。
小于b的为1，大于b的为-1。
最后整一段的和等于0那么中位数肯定是b。
用l,r。
l[i]表示从b开始向左连续一段和为i的有多少种情况。
比如l[1]等于从b这个位置开始向左连续一段和为1的有多少种情况。
r[i]就表示往右。
那么l[1]*r[-1]就是一部分的答案。
左边和为1右边和为-1加起来总和为0那么b肯定是中位数。

代码实现：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
int a[110000],l[210000],r[210000];
int main() {
    int n,b,p;scanf("%d%d",&n,&b);
    for(int i=1;i<=n;i++) {
        int x;scanf("%d",&x);
        if(x<b)
            a[i]=1;
        if(x>b)
            a[i]=-1;
        if(x==b)
            p=i;
    }
    //数组不能为负所以都加上n。
    int sum=0;l[n]=r[n]=1;  //l[n]和r[n]相当于l[0]和r[0],也就是一开始不延伸和等于0的就一种情况就是只有自己。
    for(int i=p-1;i>=1;i--) {
        sum+=a[i];l[sum+n]++;
    }
    sum=0;
    for(int i=p+1;i<=n;i++) {
        sum+=a[i];r[sum+n]++;
    }
    int ans=0;
    for(int i=-n;i<=n;i++)
        ans+=l[n-i]*r[i+n]; //其实是l[i]*r[-i],但是每个都加了n，所以成这样了。
    printf("%d\n",ans);
    return 0;
}