【题解】ABC280E Critical Hit

题目

题目大意

有一生命值为 $N$ 的怪兽，你对其攻击一次，有 $1-\frac{P}{100}$ 的概率使其扣 $1$ 点生命值，有 $\frac{P}{100}$ 的概率使其扣 $2$ 点生命值。求击杀其(指其生命值不大于 $0$)的期望攻击次数($\mathrm{m}\mathrm{o}\mathrm{d}998244353$)。

分析

首先你需要了解期望dp

显然，这是一个期望dp的板题。

设 $f[i]$ 为怪兽生命值为 $i$ 时需要击杀的期望攻击次数，明显地，$f[0]=0$。

$f[1]=1$，因为当其生命值为 $1$ 时，只要进行 $1$ 次攻击，无论是扣 $1$ 点还是 $2$ 点生命值，怪兽都会死亡。

由于有 $1-\frac{P}{100}$ 的概率扣 $1$ 点生命值，那么有 $1-\frac{P}{100}$ 的概率由 $f[i-1]$ 转移而来，也就是当怪兽生命值为 $i$ 时，有 $1-\frac{P}{100}$ 的概率有如下结果：攻击 $1$ 次扣 $1$ 点生命值，此时再攻击 $f[i-1]$ 次就能杀死怪兽。

同理，当怪兽生命值为 $i$ 时，有 $\frac{P}{100}$ 的概率有如下结果：攻击 $1$ 次扣 $2$ 点生命值，此时再攻击 $f[i-2]$ 次就能杀死怪兽。

那么有
$$f[i]=(1-\frac{P}{100})\times (f[i-1]+1)+\frac{P}{100}\times (f[i-2]+1)$$

最终输出 $f[N]$.

注意这个式子是在 $\mathrm{m}\mathrm{o}\mathrm{d}998244353$ 意义下的，所以对于分数 $1-\frac{P}{100}=\frac{100-P}{100}$ 及 $\frac{P}{100}$ 都要在模意义下运算。

我们知道，费马小定理简述如下：

$$a^{p-1}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$$
其中 $p$ 为质数。

那么有

$$a^{p-2}\equiv a^{-1}(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

那么

$$f[i]=(100-P)\times 100^{-1}\times (f[i-1]+1)+P\times 100^{-1}\times (f[i-2]+1)$$
$$=(100-P)\times 100^{998244353-2}\times (f[i-1]+1)+P\times 100^{998244353-2}\times (f[i-2]+1)$$

完毕。

code

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 200005;
const LL mod = 998244353;
int n;
LL p, f[N];
LL qpow(LL x, LL y) {
    LL sum = 1;
    while (y) {
        if (y & 1) sum = sum * x % mod;
        x = x * x % mod, y >>= 1;
    }
    return sum;
}
int main() {
    scanf("%d%lld", &n, &p);
    f[1] = 1;
    for (int i = 2; i <= n; i++)
    	f[i] = (p * qpow(100, mod - 2) % mod * (f[i - 2] + 1) % mod + (100 - p) * qpow(100, mod - 2) % mod * (f[i - 1] + 1) % mod) % mod;
    printf("%lld", f[n]);
    return 0;
}

若模数为$\mathrm{m}\mathrm{o}\mathrm{d}$，

时间复杂度为$O(N\log \mathrm{m}\mathrm{o}\mathrm{d})$