BZOJ#1303中位数图

[CQOI2009]中位数图

因为是排列，不会出现相同的数字，把比b小的赋值为1，大于b的赋值为-1，则要是b为中位数必然这一段的和为0，（注意b的位置）

代码：

const int N=200010;
int a[N];
int s[N];
void solve(int Case) {
    int n,b;
    cin>>n>>b;
    for(int i=1;i<=n;i++) cin>>a[i];
    int p=-1;
    for(int i=1;i<=n;i++){
        if(a[i]<b) s[i]=1;
        else if(a[i]>b) s[i]=-1;
        if(a[i]==b) p=i;
    }
    for(int i=1;i<=n;i++) s[i]+=s[i-1];
    map<int,int> mp;
    mp[0]++;
    int res=0;
    for(int i=1;i<=n;i++){
        if(i>=p){
            res+=mp[s[i]];
        }
        if(i<p) mp[s[i]]++;
    }
    cout<<res<<nline;
}

标签：1303,int,res,中位数,mp,赋值,BZOJ
来源： https://www.cnblogs.com/koto-k/p/16072024.html