【BZOJ】1303 [CQOI2009]中位数图 建模

题目传送门

这题最主要的思想就是建模，对于每一个数a[i]，建立一个c数组：

1. 若a[i]<b，则c[i]=-1。
2. 若a[i]=b，则c[i]=0，用一个变量wz记录当前的i。
3. 若a[i]>b，则c[i]=1。

这样我们就把题目转化成了求c数组里有多少子序列的和为0。

之后我们再对c数组做一遍前缀和，然后枚举两个位置i和j，设i<j，若c[i]=c[j]，则答案+1。

因此，我们设k=0->wz-1，统计每个c[k]的出现次数，然后再设k=wz->n，ans+=c[k]出现的次数。

最后的ans就是答案咯。

注意：为了防止c[k]为下标时c[k]<0，不符合C++语言，因此我们可以把c[k]+=n。

附上AC代码：

#include <cstdio>using namespace std;int n,m,x,c[100010],t[200010],ans,wz;int main(void){scanf("%d%d",&n,&m);for (int i=1; i<=n; ++i){scanf("%d",&x);if (x<m) c[i]=-1;if (x==m) c[i]=0,wz=i;if (x>m) c[i]=1;c[i]+=c[i-1];}for (int i=0; i<wz; ++i) ++t[c[i]+n];for (int i=wz; i<=n; ++i) ans+=t[c[i]+n];printf("%d",ans);return 0;}