【CQOI2009】中位数

Description

给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后，位于中间的数。

Solution

好久没有见到这么短的题目了。
一看到维护中位数b就要想到把＜b的变为-1，＞b的变为1，然后若有一段区间和为0，那么这一段的区间的中位数就是b了。
设b的位置为j。
那么很明显打一个存值域的桶就好了，设sum[i]为和是i的区间的有多少个（必须末尾为j或开头为j），然后用l[i]表示末尾为j的区间和为i的有多少个，r[i]表示开头为j的区间和为i的有多少个。
存在负数，c++要怎么办？把所有的值加上一个n就好了。
最后答案每次就加上一个l[i]*r[2*n-i]。
注意l[n]=r[n]=1（所有的值域都是加上n的，所以原意就是l[0]=r[0]=1）

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int maxn=200007;
int i,j,k,t,n,m,b,a[maxn];
long long ans;
int l[maxn],r[maxn],sum[maxn];
int main(){
    scanf("%d%d",&n,&b);
    fo(i,1,n){
        scanf("%d",&k);
        if(k<b)a[i]=-1;else if(k>b)a[i]=1;
        if(k==b){
            j=i;
        }
    }
    l[n]=r[n]=1;
    fod(i,j-1,1){
        sum[i]=sum[i+1]+a[i];
        l[n+sum[i]]++;
    }
    fo(i,j+1,n){
        sum[i]=sum[i-1]+a[i];
        r[sum[i]+n]++;
    }
    fo(i,0,2*n)ans+=l[i]*r[2*n-i];
    printf("%lld\n",ans);
}