BZOJ_P1303 [CQOI2009]中位数图(中位数)

BZOJ传送门

Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 1974 Solved: 1268
[Submit][Status][Discuss]
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

Source

Sol:
将在b前面大于或小于b的个数统计，用前缀和
在前面大于(或小于)b的与在后面小于(或大于)b的正好抵消，乘一下就可以

#include<cstdio>
#define N 100005
int l[N<<1],r[N<<1],a[N];int n,b,k,L,R,ans;
inline int in(int x=0,char ch=getchar()){while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x;}
int main(){
    n=in(),b=in();
    for(int i=1;i<=n;i++){a[i]=in();if(a[i]==b) k=i,a[i]=0;else a[i]=(a[i]>b?1:-1);}
    for(int i=k;i;i--) L+=a[i],l[L+n]++;
    for(int i=k;i<=n;i++) R+=a[i],r[R+n]++;
    for(int i=0;i<n*2;i++) ans+=l[i]*r[n*2-i];
    printf("%d\n",ans);
    return 0;
}