bzoj1303 [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\leqslant 100000$

Solution

显然要求子序列中大于$b$的个数等于小于$b$的个数。用前缀和，小于$b$的记为$-1$，大于$b$的记为$1$。$l[i]$表示左边前缀和为$i$的位置的个数，$r[i]$表示右边前缀和为 i <script type="math/tex" id="MathJax-Element-363">i</script>的位置的个数。注意负数处理。

#include<bits/stdc++.h>
using namespace std;

inline int read() {
    int x = 0, flag = 1; char ch = getchar();
    while (ch > '9' || ch < '0') { if (ch == '-') flag = -1; ch = getchar(); }
    while (ch <= '9' && ch >= '0') { x = x * 10 + ch - '0'; ch = getchar(); }
    return x * flag;
}
inline void write(int x) { if (x >= 10) write(x / 10); putchar(x % 10 + '0'); }

#define N 100001
#define rep(i, a, b) for (int i = a; i <= b; i++)
#define drp(i, a, b) for (int i = a; i >= b; i--)

int n, b, a[N];
int num[N];
int l[N << 1], r[N << 1];

int main() {
    cin >> n >> b;
    int pos = 0, sum, ans;
    rep(i, 1, n) {
        a[i] = read();
        if (a[i] == b) pos = i, num[i] = 0;
        else if (a[i] < b) num[i] = -1;
        else num[i] = 1;
    }
    l[n] = r[n] = 1;
    sum = 0; drp(i, pos - 1, 1) sum += num[i], l[sum + n]++;
    sum = 0; rep(i, pos + 1, n) sum += num[i], r[sum + n]++;
    ans = 0; rep(i, 0, (n << 1) - 1) ans += l[i] * r[2 * n - i];
    write(ans);
    return 0;
}