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

一开始是想从中位数往两边扩展
后来证伪了……
我们发现数的贡献只和其与中位数大小有关，和其本身是无关的
我们把小于中位数的值设为-1，大于的设为1
显然若区间[l,r]包含中位数且值和为0，那么就是满足条件的
（为什么能保证是奇数长？因为偶数不可能加出和为0的……）
若要判断[l,r]区间，需要sum[r]-sum[l-1]=0,即sum[r]=sum[l-1]
我们将[0,pos-1]的sum值放到一个桶里，然后枚举右端点统计答案即可。

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<map>
 #define N (100000+100)
 using namespace std;

 int n,m,t,a[N],mid,sum[N];
 long long ans;
 map<int,int>Keg;

 int main()
 {
     scanf("%d%d",&n,&m);
     for (int i=; i<=n; ++i)
     {
         scanf("%d",&a[i]);
         if (a[i]==m)  t=i,a[i]=;
         else  a[i]=(a[i]<m)?-:;
         sum[i]=sum[i-]+a[i];
         if (!t)  Keg[sum[i]]++;
     }
     Keg[]++;//注意这里
     for (int i=t; i<=n; ++i)
         ans+=Keg[sum[i]];
     printf("%lld",ans);
 }