计蒜客 A1617 Hall 定理 + 尺取法 + 线段树

题意

传送门 计蒜客 A1617 LOVER II

题解

先不考虑区间查询。对于 $b$ 与 $a$，令其升序排序。令 $c[i]$ 等于满足 $a[i]+b[j]\ge k$ 的 $j$ 的数量。$c[i]$ 单调不减，且满足条件的 $j$ 是 $a$ 的后缀。满足条件当且仅当 $c[i]\ge i$。可以通过 Hall 定理证明，满足上述条件时存在 $a$ 的完备匹配。

对于 $b$，若 $l$ 增加，满足条件的 $r$ 一定不减。那么可以使用尺取法求解，同时使用可区间修改的线段树维护 $c[i]-i$，若 $[1,n]$ 最小值大于等于零，则满足条件。总时间复杂度 $O(n\log n)$。

#include <bits/stdc++.h>
using namespace std;
constexpr int MAXN = 2E5 + 5, SZ = 1 << 19;
int T, N, M, K, Q;
int A[MAXN], B[MAXN];
int F[MAXN];

struct ST
{
    int dat[SZ], lz[SZ];
    void init(int k = 0, int l = 0, int r = N)
    {
        if (r - l == 1)
        {
            dat[k] = -(l + 1);
            return;
        }
        int m = (l + r) / 2, chl = k * 2 + 1, chr = k * 2 + 2;
        init(chl, l, m), init(chr, m, r);
        lz[k] = 0;
        dat[k] = min(dat[chl], dat[chr]);
    }
    void pushdown(int k)
    {
        if (lz[k] != 0)
        {
            int chl = k * 2 + 1, chr = k * 2 + 2;
            int x = lz[k];
            lz[k] = 0;
            lz[chl] += x, dat[chl] += x;
            lz[chr] += x, dat[chr] += x;
        }
    }
    void change(int a, int b, int x, int k = 0, int l = 0, int r = N)
    {
        if (r <= a || b <= l)
            return;
        if (a <= l && r <= b)
        {
            lz[k] += x, dat[k] += x;
            return;
        }
        pushdown(k);
        int m = (l + r) / 2, chl = k * 2 + 1, chr = k * 2 + 2;
        change(a, b, x, chl, l, m), change(a, b, x, chr, m, r);
        dat[k] = min(dat[chl], dat[chr]);
    }
} tr;

int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin >> T;
    while (T--)
    {
        cin >> N >> M >> K;
        for (int i = 0; i < N; ++i)
            cin >> A[i];
        for (int i = 0; i < M; ++i)
            cin >> B[i];
        sort(A, A + N);
        tr.init();
        for (int l = 0, r = 0; l < M;)
        {
            while (r < M && tr.dat[0] < 0)
            {
                int pos = lower_bound(A, A + N, K - B[r]) - A;
                if (pos < N)
                    tr.change(pos, N, 1);
                ++r;
            }
            F[l] = tr.dat[0] >= 0 ? r : M + 1;
            int pos = lower_bound(A, A + N, K - B[l]) - A;
            if (pos < N)
                tr.change(pos, N, -1);
            ++l;
        }
        cin >> Q;
        while (Q--)
        {
            int l, r;
            cin >> l >> r;
            --l;
            cout << (F[l] <= r ? 1 : 0) << '\n';
        }
    }
    return 0;
}