【洛谷】P8774 [蓝桥杯 2022 省 A] 爬树的甲壳虫 的题解

【洛谷】P8774 [蓝桥杯 2022 省 A] 爬树的甲壳虫 的题解

题目传送门

思路

先考虑求从 $0$ 到 $1$ 的期望。

有 $p_1$ 的概率掉落，$1-p_1$ 的概率向上移动一次。

只走一次上去的期望时间是 $1\times (1-p_1)$。

只走两次上去的期望时间是 $2\times (1-p_1)\times p_1$。

只走三次上去的期望时间是 $3\times (1-p_1)\times p_1^2$。

所以期望时间是 $E_x(1)={\sum }_{i=1}i\times (1-p_1)\times p_1^{i-1}=(1-p_1)\times {\sum }_{i=1}i\times p_1^{i-1}$。

令 $F(x)={\sum }_{i=1}i\times x^{i-1}$ 则 $F(x)={\sum }_{i=1}(x^i{)}^{{}^{\prime }}=({\sum }_{i=1}{x}^i{)}^{{}^{\prime }}=({\sum }_{i=0}{x}^i{)}^{{}^{\prime }}$。

显然，$F(x)$ 是首项为 $x$ 的等比数列求和，运用等比数列求和公式可得 $F(x)=(\frac{1-x^{\infty }}{1-x}{)}^{{}^{\prime }}$ 因为 $x<1$，所以 $F(x)=(\frac{1}{1-x}{)}^{{}^{\prime }}=\frac{1}{(1-x{)}^2}$。

所以，$E_x(1)=(1-p_1)\times F(p_1)=(1-p_1)\times \frac{1}{(1-p_1{)}^2}=\frac{1}{1-p_1}$。

那么从 $n-1$ 到 $n$ 的期望时间分析如下：

• 一次上去的期望时间是 $E_x(n-1)+1)\times (1-p_n)$

• 两次上去的期望时间是 $2\times (E_x(n-1)+1))\times (1-p_n)\times p_n$ (理解为第一次掉下去，第二次上去)。

• 三次上去的期望时间是 $3\times (E_x(n-1)+1))\times (1-p_n)\times p_n^2$ (理解为前两次掉下去，第三次上去)。

那么，同理可得 $E_x(n)=(E_x(n-1)+1)\times (1-p_n)\times {\sum }_{i=1}i\times p_n^{i-1}=(E_x(n-1)+1)\times \frac{1}{1-p_n}$。

即得到递推式 $E_x(n)=(E_x(n-1)+1)\times \frac{1}{1-p_n}$。

即 $E_x(n)=(E_x(n-1)+1)\times \frac{y_i}{y_i-x_i}$。

对于除以 $(y_i-x_i)$ 在 $\mathrm{mod}P$ 的意义下等价于乘以 $(y_i-x_i{)}^{-1}$，由于 $P$ 为质数，根据费马小定理，运用快速幂求逆元。

代码

#include <bits/stdc++.h>
#define endl "\n"
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
namespace fastIO {
	inline int read() {
		register int x = 0, f = 1;
		register char c = getchar();
		while (c < '0' || c > '9') {
			if(c == '-') f = -1;
			c = getchar();
		}
		while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
		return x * f;
	}
	inline void write(int x) {
		if(x < 0) putchar('-'),x = -x;
		if(x > 9) write(x / 10);
		putchar(x % 10 + '0');
		return;
	}
}
using namespace fastIO;
const ll Mod = 998244353ll;
int n;
ll ans = 0;
ll fast_pow(ll x, ll y) {
	ll re = 1ll;
	while(y) {
		if(y & 1ll) re = re * x % Mod;
		x = x * x % Mod;
		y >>= 1;
	}
	return re;
}
int main() {
	//freopen(".in","r",stdin);
	//freopen(".out","w",stdout);
	ll x, y;
	n = read();
	for(ll i = 1ll; i <= n; i ++) {
		scanf("%lld%lld", &x, &y);
		ans = (ans + 1ll) * y % Mod * fast_pow(y - x, Mod - 2ll) % Mod;
		ans = (ans + Mod) % Mod;
	}
	printf("%lld", ans);
	return 0;
}