[CQOI2009] 中位数
[CQOI2009] 中位数
BZOJ
luogu
考虑\(n^2\)的暴力怎么写,对每个点记前缀和f,g表示1~i大于和小于b的个数,
枚举左右端点(区间要包含b所在的点),查询这段区间内大于b和小于b的个数,相等就计入答案
写完暴力正解就出来了
考虑只枚举左端点,我们要求有多少右端点满足条件
暴力的式子\(f[r]-f[l-1]=g[r]-g[l-1]\),那么直接开桶记录每个点的f-g,直接查就好了
#include<bits/stdc++.h>
using namespace std;
const int _=2e5+5;
int re(){
int x=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return x*w;
}
int n,k,b,f[_],g[_],t[_];
long long ans;
int main(){
n=re(),b=re();
for(int i=1,a;i<=n;i++){
a=re();f[i]=f[i-1];g[i]=g[i-1];
if(a>b)f[i]++;
else if(a<b)g[i]++;
else k=i;
}
for(int i=k;i<=n;i++)t[g[i]-f[i]+n]++;
for(int i=0;i<k;i++)ans+=t[g[i]-f[i]+n];
printf("%lld\n",ans);return 0;
}