[CQOI2009]中位数图(数学转换+思维+前缀后缀和+统计+扩大区间)

题意：传送门

题解：这是个很好的题，首先找到这个b位置后，然后把这个位置左右两边都可以进行转化，如果大于这个b，那么转化为1，小于这个b，转化成-1，然后从这个位置开始向左求个前缀和，lsum[i,pos]表示从pos这个位置到i这个位置的和，从这个位置开始向右求个前缀和rsum[i,point]表示从point到i这个位置的和，之后在lsum[i,pos]统计出现的个数lmapp[lsum[i,pos]]，在rsum[i,pos]统计出现的个数rmapp[rsum[i,point]]，然后就可以乘开了，但是注意负数要扩大区间处理下，之后ans+=lmapp[lsum[i,pos]]*rmapp[-lsum[i,pos]]就可以得出答案了。

附上代码：

#include<bits/stdc++.h>

using namespace std;

const int maxn=1e5+5;

int n,b,pos,ans;
int num[maxn],sum[maxn],l[maxn*2],r[maxn*2];

int main()
{
    scanf("%d%d",&n,&b);
    for(int i=1;i<=n;i++){
        scanf("%d",&num[i]);
        if(num[i]>b)num[i]=1;
        else if(num[i]==b){
            num[i]=0;pos=i;
        }else num[i]=-1;
    }
    l[n]=1;r[n]=1;
    for(int i=pos-1;i>=1;i--){
        sum[i]=sum[i+1]+num[i];l[sum[i]+n]++;
    }
    for(int i=pos+1;i<=n;i++){
        sum[i]=sum[i-1]+num[i];r[sum[i]+n]++;
    }
    for(int i=0;i<2*n-1;i++){
        ans+=l[i]*r[2*n-i];
    }
    printf("%d\n",ans);
    return 0;
}