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;
}