GDKOI 2021 TG DAY2 游戏（概率 dp）（解析+代码）（本文读者绝对 rp +++++）

本文链接：https://blog.csdn.net/L_Aria/article/details/141430830

GDKOI 2021 TG DAY2 游戏（概率 dp）（解析+代码）（本文读者绝对 rp +++++）

序

貌似是做的第一道概率 dp 的题目

还是蛮有意思的

题意

当前有 $0$ 颗星，当持有星数为 $i$ 时，有 $p_i$ 的概率可以获得一颗星，否则扣除一颗星。问期望需要多少次对局能获得 $n$ 颗星，输出答案 $%p\%p$ ， $p = 998244353$ 。

解法

概率 dp 走起

我们设 $f_i$ 表示从 $i - 1$ 走到 $i$ 的期望局数

回到题目，我们尝试推出转移方程：

如果成功，那么耗费步数 $1$

如果失败，那么将会掉到 $i - 2$ ，那么再回到 $i$ 的概率就是 $f_{i-1}+f_i$ ，要乘上失败的概率 $1-p_i)$

那么转移为 $f_i=1+(f_{i-1}+f_i)(1-p_i)$

发现一个有趣的问题

为啥自己转移自己？

拆一下就好啦
$f_i=\frac{f_{i-1}(1-p_i)+1}{p_i}$

依此递推即可。

Q：可是概率是个分数吖，肿么办？

A：众所周知， $i$ 的逆元 $in v (i)$ 就是在 $%p\% p$ 意义下， $i$ 的倒数，因此，除以一个数也就等于乘以它的逆元。

Q：什么，逆元！？ exgcd 不要啊！

A：这道题的模数 $998244353$ 是一个质数，根据费马小定理，当模数为质数 $p$ 时， $i$ 的逆元就是 $i^{p-2}$ ，快速幂即可。

鉴于我们状态的含义，最后的答案是 $ans=∑i=1nfians=\sum_{i=1}^{n} f_i$

不要忘记取模吖

代码

#include<cstdio>
#define ll long long int 
using namespace std;
const ll Md=998244353;
const int N=1e6;
ll a[N+10],b[N+10];
ll inv(ll x)
{
	ll y=Md-2;
	ll ans=1;
	while(y)
	{
		if(y&1)
		{
			ans=(ans*x)%Md;
		}
		x=(x*x)%Md;
		y>>=1;
	}
	return ans;
}
int main()
{
	freopen("game.in","r",stdin);
	freopen("game.out","w",stdout);
	int n;
	scanf("%d",&n);
	ll ans=0,lst=0;
	for(int i=1;i<=n;i++)
	{
		ll x,y;
		scanf("%lld%lld",&x,&y);
		ll t=inv(y),t1=(x*t)%Md,t2=((y-x)*t)%Md;
		lst=(((lst*t2)%Md+1)%Md*inv(t1))%Md;
		ans=(ans+lst)%Md;
	}
	printf("%lld",ans);
	return 0;
}