BZOJ 1303 中位数图 题解

题面

因为所求的是中位数，所以考虑改变原序列。把大于 b 的数全部变为 1，小于 b 的数变为 −1，等于 b 则为 0。问题就变为求存在几个包含 b的区间和为 0 。

根据乘法原理，我们枚举每一个l[i]，求出l[i]*r[-i]后累加到ans上；

因为数组的下标不能是负数，所以把所有下标都加上100000；

#include <bits/stdc++.h>
using namespace std;
int a[];
int c[];
int r[],l[];
int mmap1[],mmap2[];
int main()
{
    int n,b;
    cin>>n>>b;
    int id=;
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
        if(a[i]==b) id=i;
        if(a[i]>b) c[i]=;
        else if(a[i]<b) c[i]=-;
        else c[i]=;
    }
    for(int i=id;i<=n;i++){
        r[i]=r[i-]+c[i];
        mmap2[r[i]+]++;
    }
    for(int i=id;i>=;i--){
        l[i]=l[i+]+c[i];
        mmap1[l[i]+]++;
    }
    long long ans=;
    for(int i=;i<=;i++){
        ans+=mmap1[i]*mmap2[-i];
    }
    cout<<ans;
}