BZOJ 1303: [CQOI2009]中位数图

刚开始以为有多个b,自闭了好一会儿

从b所在那个位置开始，分别往左往右

定义一个临时变量$tmp$

遇到$>b 的 就+1, 遇到<b的就-1$

$那么左边的tmp 肯定能够和 右边的-tmp 匹配使得这一段区间中b是中位数$

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

#define ll long long
#define N 100010
int n, b, a[N];
int cntl[N << 1], cntr[N << 1];

void Run()
{
    while (scanf("%d%d", &n, &b) != EOF)
    {
        int pos = -1;
        for (int i = 1; i <= n; ++i)
        {
            scanf("%d", a + i);
            if (a[i] == b) pos = i;
        }
        for (int i = pos - 1, tmp = 0; i >= 1; --i)
        {
            if (a[i] > b) ++tmp;
            else if (a[i] < b) --tmp;
            ++cntl[tmp + N];
        }
        for (int i = pos + 1, tmp = 0; i <= n; ++i)
        {
            if (a[i] > b) ++tmp;
            else if (a[i] < b) --tmp;
            ++cntr[tmp + N];
        }
        ll res = (cntl[N] + 1) * (cntr[N] + 1);
        for (int i = 1; i <= n; ++i)
            res += cntl[N + i] * cntr[N - i] + cntl[N - i] * cntr[N + i];
        printf("%lld
", res);
    }
}

int main()
{
    #ifdef LOCAL
        freopen("Test.in", "r", stdin);
    #endif 

    Run();
    return 0;
}