题意:

我们记一个长度为 $n$ 的序列的中位数是它排序后 $\lfloor \frac{n}{2} \rfloor+1$ 位置上的数。

现在有一个长度为 $n$ 的序列 $a$,对于任意 $1\le l \le r \le n$,我们取出区间 $[l,r]$ 的中位数,记为 $b[l][r]$。

求 $b$ 的中位数。 $n \le 100000$


思路:

首先我们求中位数有一个很套路的方法:

  • 我们二分一个数 $mid$,对于原序列中 $\ge mid$ 的数,我们标记为 $1$;反之,对于 $< mid$ 的数,我们标记为 $-1$。

  • 标记结束后,我们对所有标记求和,记为 $sum$。如果 $sum \ge 0$,说明中位数是 $\ge mid$ 的,那么向右继续二分;反之,向左二分。

对于这题,我们思路一样是二分一个数 $mid$,并对原序列进行标记。

我们的问题就转化为:统计有多少个区间,其标记和 $\ge 0$。

记满足条件的区间有 $cnt$ 个,如果 $cnt\ge \lfloor \frac{n(n+1)/2+1}{2} \rfloor$,那么说明 $ans\ge mid$,继续向右二分;反之向左二分。

现在来思考怎么计算这个 $cnt$。我们考虑对标记做一个前缀和,设为 $s$。

一个满足条件的区间 $[l,r]$ 需要 $s_r-s_{l-1}$。

这是一个经典的计数问题,用树状数组维护。

时间复杂度为 $O(n\log^2n)$


代码

#include<bits/stdc++.h>
#define LL long long
#define FOR(i, x, y) for(int i = (x); i <= (y); i++)
#define ROF(i, x, y) for(int i = (x); i >= (y); i--)
using namespace std;
inline int rd()
{
    int sign = 1, re = 0; char c = getchar();
    while(!isdigit(c)){if(c == '-') sign = -1; c = getchar();}
    while(isdigit(c)) {re = re * 10 + (c - '0'); c = getchar();}
    return sign * re;
}
int n, a[100005], b[100005], s[100005];
LL tot;
int tr[200005];
inline int lb(int x) {return x & (-x);}
inline void add(int x, int val)
{
    while(x <= (n << 1) + 1)
    {
        tr[x] += val;
        x += lb(x);
    }
}
inline int query(int x)
{
    int re = 0;
    while(x)
    {
        re += tr[x];
        x ^= lb(x);
    }
    return re;
}
inline bool chk(int mid)
{
    LL re = 0;
    FOR(i, 1, n)
        if(a[i] < s[mid]) b[i] = -1;
        else b[i] = 1;
    FOR(i, 1, n)
    {
        b[i] += b[i - 1];
        re += query(b[i] + n + 1);
        if(b[i] >= 0) re++;
        add(b[i] + n + 1, 1);
    }
    ROF(i, n, 1) add(b[i] + n + 1, -1);
    return re >= tot;
}
signed main()
{
    n = rd(); tot = 1ll * n * (n + 1) / 2ll; tot = (tot + 1) / 2ll;
    FOR(i, 1, n) a[i] = s[i] = rd();
    sort(s + 1, s + 1 + n);
    int l = 0, r = n + 1;
    while(l + 1 < r)
    {
        int mid = (l + r) >> 1;
        if(chk(mid)) l = mid;
        else r = mid;
    }
    printf("%d", s[l]);
    return 0;
}