Address
Solution
-
神仙题 QAQ
-
考虑非 DP 的做法。
-
为了简化问题,我们用二元组 ( x , d ) (x,d) (x,d) 表示两点中坐标数值更大的点在 x x x 且坐标数值更小的点到坐标数值更大的点的距离为 d d d,每次有以下三种操作:
- 将 x x x 和 d d d 同时加 1 1 1;
- 若 d ≥ 2 d \ge 2 d≥2,将 d d d 减 1 1 1;
- 将 d d d 变为 0 0 0。
-
先考虑只进行
操作1
和操作2
的方案数,枚举进行操作1
和操作2
的总次数,过程中要保证 d d d 始终大于 0 0 0,用类似卡特兰数的计算方法计算即可。 -
令只进行
操作1
和操作2
之后坐标数值较小的点的坐标为 k k k。 -
若
操作3
的数量为 0 0 0,方案合法当且仅当 k = A k = A k=A。 -
若
操作3
的数量大于 0 0 0,考虑将操作3
插入到操作1
和操作2
的序列中。 -
将序列中第 i i i 次操作后 d d d 的值记作 d i d_i di,可以发现
操作3
能够插在第 i i i 次操作之后当且仅当不存在 j ( i < j ) j(i < j) j(i<j) 使得 d j ≤ d i d_j \le d_i dj≤di。 -
为了让最后的 d d d 为 B − A B - A B−A,最后一次
操作3
一定是插在 d d d 最后一次为 A − k A - k A−k 的操作之后,那么剩余的操作3
就可以插在 d d d 最后一次为 0 ∼ A − k 0 \sim A - k 0∼A−k 的任意一个操作之后,用插板法计算即可。 -
时间复杂度 O ( n ) \mathcal O(n) O(n)。
Code
#include <bits/stdc++.h>
const int N = 1e7 + 5;
const int mod = 998244353;
int fra[N], inv[N];
int n, a, b, ans;
template <class T>
inline T Min(T x, T y) {return x < y ? x : y;}
inline void add(int &x, int y)
{
x += y;
x >= mod ? x -= mod : 0;
}
inline void dec(int &x, int y)
{
x -= y;
x < 0 ? x += mod : 0;
}
inline void add_prod(int &x, int u, int v)
{
x = (1ll * u * v + x) % mod;
}
inline int quick_pow(int x, int k)
{
int res = 1;
while (k)
{
if (k & 1)
res = 1ll * res * x % mod;
x = 1ll * x * x % mod;
k >>= 1;
}
return res;
}
inline int C(int n, int m)
{
if (n < m)
return 0;
return 1ll * fra[n] * inv[n - m] % mod * inv[m] % mod;
}
inline int ask(int k)
{
int res = C(k + b - 1, b - 1);
dec(res, C(k + b - 1, k - 1));
return res;
}
int main()
{
scanf("%d%d%d", &n, &a, &b);
if (a == 0 && b == 0)
return puts("1"), 0;
fra[0] = 1;
for (int i = 1; i <= n; ++i)
fra[i] = 1ll * fra[i - 1] * i % mod;
inv[n] = quick_pow(fra[n], mod - 2);
for (int i = n; i >= 1; --i)
inv[i - 1] = 1ll * inv[i] * i % mod;
for (int k = 0, km = Min(n - b, Min(a, b - 1)); k <= km; ++k)
{
int tmp = k ? ask(k) : 1;
if (n - b - k == 0)
{
if (k == a)
add(ans, tmp);
}
else
{
int x = n - b - k - 1,
y = a - k + 1;
if (x >= 0)
add_prod(ans, tmp, C(x + y - 1, y - 1));
}
}
printf("%d\n", ans);
return 0;
}