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;}