题意
题解
对于每一个 j ( 1 ≤ j ≤ n − l e n + 1 ) j(1\leq j\leq n-len+1) j(1≤j≤n−len+1),求是否存在一个次序使 a [ j + i ] = b [ i ] , 1 ≤ i ≤ l e n a[j+i] = b[i],1\leq i\leq len a[j+i]=b[i],1≤i≤len。
对于 l e n len len 长的 b b b 与 a a a 的子数组,令其升序排序。令 c [ i ] c[i] c[i] 等于满足 b [ i ] + a [ j ] ≥ h b[i] + a[j] \geq h b[i]+a[j]≥h 的 j j j 的数量。 c [ i ] c[i] c[i] 单调不减,且满足条件的 j j j 是一个 a a a 子数组的后缀。满足条件当且仅当 c [ i ] ≥ i c[i]\geq i c[i]≥i。有两种证明策略。
一种是贪心策略。对于 b b b,依次考虑 i i i,贪心地取可以选取的,且未被 i ′ ( i ′ < i ) i^\prime(i^\prime<i) i′(i′<i) 选取的 A [ j ] A[j] A[j]。若对于所有 i i i 都能找到匹配的 j j j,则有 c [ i ] ≥ i c[i]\geq i c[i]≥i。
一种是根据 Hall 定理证明。问题等价于求完备匹配。若 b [ i ] + a [ j ] ≥ h b[i] + a[j] \geq h b[i]+a[j]≥h,看作两个节点之间连接了一条无向边。对任一个 b b b 的规模为 k k k 子集 S S S,当且仅当与其连边的点集 A ( S ) A(S) A(S) 规模满足 ∣ A ( S ) ∣ ≥ ∣ S ∣ \lvert A(S) \rvert \geq \lvert S\rvert ∣A(S)∣≥∣S∣,二分图中存在完备匹配。由于 j > i j>i j>i 时, b [ j ] b[j] b[j] 连边的点集包含 b [ i ] b[i] b[i] 连边的点集,故只用考虑点集中索引最大者即可。又由于 c [ i ] c[i] c[i] 单调不减,则条件为 c [ i ] ≥ i c[i]\geq i c[i]≥i。
令 f [ i ] = c [ i ] − i f[i] = c[i] - i f[i]=c[i]−i,维护可区间修改的线段树。总时间复杂度 O ( n log n ) O(n\log n) O(nlogn)。
#include <bits/stdc++.h>
using namespace std;
constexpr int MAXN = 15E4 + 5, SZ = 1 << 19;
int N, L, H;
int A[MAXN], B[MAXN];
struct ST
{
int dat[SZ], lz[SZ];
void init(int k = 0, int l = 0, int r = L)
{
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 = L)
{
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 >> N >> L >> H;
for (int i = 0; i < L; ++i)
cin >> B[i];
for (int i = 0; i < N; ++i)
cin >> A[i];
sort(B, B + L);
tr.init();
for (int i = 0; i < L; ++i)
{
int pos = lower_bound(B, B + L, H - A[i]) - B;
if (pos < L)
tr.change(pos, L, 1);
}
int res = tr.dat[0] >= 0;
for (int i = L; i < N; ++i)
{
int pos = lower_bound(B, B + L, H - A[i]) - B;
if (pos < L)
tr.change(pos, L, 1);
pos = lower_bound(B, B + L, H - A[i - L]) - B;
if (pos < L)
tr.change(pos, L, -1);
res += tr.dat[0] >= 0;
}
cout << res << '\n';
return 0;
}