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

#include<cstdio>
#include<iostream>
using namespace std;
int n,m,id,s1,s2;
long long ans;
int a[100005],p1[200005],p2[200005];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) 
    {
        scanf("%d",&a[i]);
        if(a[i]==m) id=i;
    }
    p1[0+n]=1;
    for(int i=1;i<id;i++) 
    {
        if(a[i]>m) s1++;
        if(a[i]<m) s2++;
        p1[s1-s2+n]++;
    }
    for(int i=id;i<=n;i++) 
    {
        if(a[i]>m) s1++;
        if(a[i]<m) s2++;
        p2[s1-s2+n]++;
    }
    for(int i=0;i<=2*n;i++) ans=ans+(long long)(p1[i])*p2[i];
    cout<<ans;
    return 0;
}