BZOJ 1303 [CQOI2009]中位数图

Description

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

Input

第一行为两个正整数n和b ，第二行为1~n 的排列。

Output

输出一个整数，即中位数为b的连续子序列个数。

Sample Input

7 4
5 7 2 4 3 1 6

Sample Output

4

HINT

第三个样例解释：{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000

传送门

好题好题。。巧妙的方法……差一点差一点= =

首先可以转化一下问题，设比b大的数都看作1，比b小的都看作-1，b看作0，

那么对于一段L~R，所有数的和为0的时候，b一定是中位数。

所以对新序列做一个前缀和sum[i]，然后就是判断sum[R]-sum[L-1]=0，

也就是sum[R]=sum[L-1]

当然还要满足L<=pos(b)，R>=pos(b)，其中pos(b)是b在原序列里的位置。

这样子的L，R一定是奇数长度的，

……很显然吧，因为当包含b的时候，b没有影响，其它-1和1的个数一定一致，

那么个数相当于2n+1。

那么只要x=1~n枚举，如果x<pos(b)就建一个哈希表统计x出现次数++，

如果x>pos(b)直接加上x之前出现的次数即可。

#include<bits/stdc++.h>
using namespace std;
int n,b;
int a[100005],num[200005];
int main(){
	scanf("%d%d",&n,&b);
	int pos,ans=0;
	for (int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		if (a[i]<b) a[i]=-1;
		if (a[i]>b) a[i]=1;
		if (a[i]==b) a[i]=0,pos=i;
	}
	for (int i=1;i<=n;i++) a[i]+=a[i-1];
	num[n]++;
	for (int i=1;i<=n;i++){
		if (i<pos) num[a[i]+n]++;
			else ans+=num[a[i]+n];
	}
	printf("%d\n",ans);
	return 0;
}