洛谷——P1627 [CQOI2009]中位数

P1627 [CQOI2009]中位数

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

中位数的题目有关统计的话，可以转化成$0,-1,1$这三个数来做，分别表示这个数$=x$，$<x$和$>x$

这个题也就是要求统计$x$左边和右边相同的$s$的对数，$s$表示前缀和之差

基数排序

#include<bits/stdc++.h>

using namespace std;

int n,x;
long long ans,c[][];

int main() {
    scanf("%d%d",&n,&x);
    int flg=;
    c[][n]=;
    for(int s=n,p,i=; i<=n; i++) {
        scanf("%d",&p);
        if(x==p) flg=;
        else s+=p>x?:-;
        c[flg][s]++;
    }
    for(int i=; i<=n*; i++)
        ans+=c[][i]*c[][i];
//        cout<<c[0][i]<<" "<<c[1][i]<<"\n";

    printf("%lld\n",ans);

    return ;
}