bzoj 1303 中位数

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=1303

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

明显卡暴力

所以用O(n)的方法

将小于b的标记为-1，大于b的标记为1，b本身标记为0

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