题意:你要从0跳到L,每一步距离不能小于d,并且下面有m组(t,p)代表一次攻击,你不能在第t次调到位置p。求方案数。()
题解:首先不考虑m次攻击,可以用dp求出答案: 。记录一下前缀和就能在 的时间范围内求出答案,并且对于1~L之间的每个距离我们都能直接得到答案(方便计算答案)。
再考虑容斥减去不可行的方案。把攻击排序,这样排在后面的攻击一定不会影响排在前面的。 表示以第n个攻击结尾,并且历经偶数次攻击; 则对应奇数次攻击。
注意到对于从0走到某一次攻击 , 可以考虑用隔板法求出答案:
并且显然地,从第i个到第j的个数,就相当于从0到
于是得到转移方程,总复杂度
#include <algorithm>
#include <iostream>
#include <vector>
#define int int64_t
#define FOR(i, x, y) for (int i = (x), _##i = (y); i < _##i; ++i)
#define FORD(i, x, y) for (int i = (x), _##i = (y); i > _##i; --i)
using namespace std;
using pii = pair<int, int>;
const int mod = 998244353;
const int maxn = 1e7 + 10;
int l, d, m;
int bin(int x, int n, int MOD)
{
int ret = MOD != 1;
for (x %= MOD; n; n >>= 1, x = x * x % MOD)
if (n & 1)
ret = ret * x % MOD;
return ret;
}
int invf[maxn], fac[maxn];
void fac_inv_init(int n, int p)
{
fac[0] = 1;
FOR(i, 1, n)
fac[i] = i * fac[i - 1] % p;
invf[n - 1] = bin(fac[n - 1], p - 2, p);
FORD(i, n - 2, -1)
invf[i] = invf[i + 1] * (i + 1) % p;
}
inline int C(int n, int m)
{ // n >= m >= 0
return n < m || m < 0 ? 0 : fac[n] * invf[m] % mod * invf[n - m] % mod;
}
inline int F(int t, int p)
{
if (p <= 0)
return 0;
return C(p - d * t + t - 1, t - 1);
}
int f[maxn], g[3005][2];
int32_t main()
{
ios::sync_with_stdio(false);
cin >> l >> d >> m;
fac_inv_init(l + 1, mod);
vector<pii> pos(m);
for (auto &p : pos)
{
cin >> p.first >> p.second;
}
f[0] = 1;
int s = 0;
for (int i = 1; i <= l; i++)
{
if (i >= d)
(s += f[i - d]) %= mod;
f[i] = s;
}
sort(pos.begin(), pos.end());
int ans = f[l];
for (int i = 0; i < m; i++)
{
if (pos[i].second > l)
continue;
g[i][1] = F(pos[i].first, pos[i].second);
for (int j = 0; j < i; j++)
{
int mem = F(pos[i].first - pos[j].first, pos[i].second - pos[j].second);
for (int op = 0; op <= 1; op++)
{
g[i][op] = (g[i][op] + g[j][1 - op] * mem) % mod;
}
}
ans = (ans - g[i][1] * f[l - pos[i].second] % mod + mod) % mod;
ans = (ans + g[i][0] * f[l - pos[i].second] % mod) % mod;
}
cout << ans << endl;
}
网友评论