BZOJ 1303: [CQOI2009]中位数图(思路题)

传送门

解题思路

　　比较好想的思路题。首先肯定要把原序列转化一下，大于\(k\)的变成\(1\)，小于\(k\)的变成\(-1\)，然后求一个前缀和，还要用\(cnt[]\)记录一下前缀和每个数出现了几次，然后统计答案的时候从\(1\)循环到\(k\)，每次转化为前缀和相减即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
const int MAXN = 100005;
typedef long long LL;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?0:1,ch=getchar();}
    while(isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return f?x:-x;
}

int n,m,a[MAXN],p;
int cnt[MAXN<<1],tag;
LL ans;

int main(){
    n=rd(),m=rd();
    for(int i=1;i<=n;i++){
        a[i]=rd();
        if(a[i]>m) a[i]=1;
        else if(a[i]==m) a[i]=0,p=i;
        else a[i]=-1;
        a[i]+=a[i-1];
        if(p) cnt[a[i]+100000]++;
    }
    ans+=cnt[100000];
    for(int i=1;i<p;i++) ans+=cnt[a[i]+100000];
    printf("%lld\n",ans);
    return 0;
}