1303: [CQOI2009]中位数图

题目链接

题目大意：给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b

题解：由于题目中的要求只与相对大小有关，为了计数的方便，把每个数离散化为1,0,-1。记b的位置为point，这样的话，对于区间[l,r]，若sum(l,r)=0&&区间长度为奇数，则l,r满足题意。用point作为序列分割点，显然l∈[1,point),r∈(point,r]。用l[i]表示左序列中数值i出现的个数，r[l]同上。枚举i=-n…n,ans+=l[i]*r[-i]。最后，由于c++不滋滋负数作数组下标，需要数组偏移一下

我的收获：数学题强啊

#include <cstdio>
#include <iostream>
using namespace std;

const int M=100005;

int n,x,b,point,ans;
int a[M],sum[M],l[M*2],r[M*2];

void work()
{
    l[0+n]=1;r[0+n]=1;//特判一下b本身
    for(int i=point-1;i>=1;i--) sum[i]=sum[i+1]+a[i],l[sum[i]+n]++;
    for(int i=point+1;i<=n;i++) sum[i]=sum[i-1]+a[i],r[sum[i]+n]++;
    for(int i=-n;i<=n;i++) ans+=l[i+n]*r[-i+n];
    printf("%d\n",ans);
}

void init()
{
    scanf("%d%d",&n,&b);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        if(x>b) a[i]=1;
        else if(x<b) a[i]=-1;
        else point=i,a[i]=0;
    }
}

int main()
{
    init();
    work();
    return 0;
}