洛谷题目： P8774 [蓝桥杯 2022 省 A] 爬树的甲壳虫题解（本题较简）

题目传送门：

P8774 [蓝桥杯 2022 省 A] 爬树的甲壳虫 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

前言：

这是一道关于概率和期望的动态规划问题，解题的核心思路是通过建立状态转移方程来计算甲壳虫从树根爬到树顶所需时间的期望值。题目还是非常的简单。

#思路：

1、问题抽象与定义状态：

我们设树的高度为，甲壳虫从高度爬到高度时掉回树根的概率为。我们定义表示甲壳虫从树根（高度为0时）爬到高度为的位置所需时间的期望值。我们的目标是求出。

2、分析状态转移过程：

考虑到甲壳虫从高度爬到高度的过程，存在两种可能的情况。

1.1、成功爬到高度：

甲壳虫以的概率从高度成功爬到高度。在这种情况下，它首先需要花费的时间从树根爬到高度爬到高度的过程中掉回树根。正在这种情况下，它先花费的时间从树根爬到高度，接着花费 1 个单位时间尝试从高度爬到高度但失败掉回树根，最后还需要花费的时间重新从树根爬到搞。所以这种情况下总共花费的时间为。

3、建立状态转移方程：

1.1、根据期望的定义，期望值是所有可能结果的概率乘以对应结果的和。对于，可以列出如下方程：

1.2、对方程进行化简：

首先展开式子：

然后合并同类项：

接着我们移项就得到：

提取公因式：

最后求解：

4、处理分数取模问题：

1.1、由于题目要求结果对质数 MOD=998244353取模，而状态转移方程中涉及到分数的计算，所以需要将分数转化为整数在模意义下进行计算。

1.2、根据费马小定理，对于质数 MOD 与 MOD 互质的整数，有（mod MOD），那么（mod MOD），也就是（mod MOD）。

1.3、因此我们在计算时，可以转换为"

（mod MOD）。这里使用快速幂算法来计算，时间复杂度为。

5、初始化和最终结果：

1.1、初始化，因为甲壳虫一开始就在树根，高度为0，所需时间的期望值为0.

1.2、然后按照状态转移方程依次计算，最终就是甲壳虫从树根爬到输定所需时间的期望值，将其对 MOD取模后输出。

##代码：

cpp 复制代码

#include <bits/stdc++.h>
using namespace std;
const int M = 998244353;
long long q(long long a, long long b) {
    long long r = 1;
    while (b) {
        if (b & 1) {
            r = r * a % M;
        }
        a = a * a % M;
        b >>= 1;
    }
    return r;
}
long long m(long long a, long long b) {
    return a * q(b, M - 2) % M;
}
int main() {
    int n;
    cin >> n;

    vector<int> x(n + 1), y(n + 1);
    vector<long long> dp(n + 1, 0);
    for (int i = 1; i <= n; ++i) {
        cin >> x[i] >> y[i];
    }

    for (int i = 1; i <= n; ++i) {
        long long p = m(x[i], y[i]);
        dp[i] = (dp[i - 1] + 1) * q(1 - p + M, M - 2) % M;
    }

    cout << dp[n] << endl;

    return 0;
}

洛谷题目： P8774 [蓝桥杯 2022 省 A] 爬树的甲壳虫 题解 （本题较简）