bzoj-1303 [CQOI2009]中位数图

1303: [CQOI2009]中位数图
题目链接
Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 3433 Solved: 2127
[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

题解
若 b 为一个序列的中位数，那么次序列中大于 b 的数的个数肯定等于小于 b 的数的个数。
又因为这个序列中没有重复（只存在一个 b）。把大于 b 的数看成 1，小于 b 的数看成 -1，哈希一下就好了。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e5+5;
int a[maxn],mid,n,h[2][maxn<<1|1];
long long ans=0;
void rad(int *x)
{
    char ch=getchar();while (ch<'0'||ch>'9') ch=getchar();
    *x=0;while (ch>='0'&&ch<='9')*x=(*x<<3)+(*x<<1)+ch-48,ch=getchar();
}
int id(int x){if (x<mid) return -1;return 1;}
int main()
{
    int *pos,*i,*ed;
    rad(&n);rad(&mid);for (i=a+1,ed=a+n+1;i<ed;++i) {rad(i);if (*i==mid) pos=i;}
    int lst=maxn;++h[pos-a&1][maxn];
    for (i=pos-1,ed=a;i>ed;--i) ++h[i-a&1][lst+=*i>mid?1:-1];
    lst=maxn;ans=h[pos-a&1][maxn];
    for (i=pos+1,ed=a+n+1;i<ed;++i) ans+=h[i-a&1][lst+=*i<mid?1:-1];
    printf("%lld\n",ans);
    return 0;
}