题目描述

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所在位置
考虑一个数比x大，能对答案造成什么影响，一个数比x小，能对答案造成什么影响，然后枚举子序列长度统计答案即可

#include <iostream>
#include <cstdio>
using namespace std;
int b,n,a[400001],pos,x,sum[200001],ans;
int l[400001],r[400001];
int main()
{
    scanf("%d%d",&n,&b);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        if(x==b) pos=i,a[i]=0;
        else if(x>b)a[i]=1;
        else a[i]=-1;
    }
    l[n]=1;r[n]=1;
    for(int i=pos-1;i>=1;i--)
        sum[i]=sum[i+1]+a[i],l[sum[i]+n]++;
    for(int i=pos+1;i<=n;i++)
        sum[i]=sum[i-1]+a[i],r[sum[i]+n]++;
    for(int i=-n+n;i<=n-1+n;i++)
        ans+=l[i]*r[n-i+n];
    printf("%d",ans);
}