2021牛客暑期多校训练营4：B-Sample Game

题目链接

补题学习思路来源于:tcy今天长胖了吗和溢流眼泪

题意:$1...n$的数的生成概率为$p_i$,执行以下操作：
$1.$随机生成一个数$x$
$2.$如果$x$不小于之前生成的所有$x$,进行$1$步骤,否则进入$3$步骤
$3.$进行分数结算,分数为生成的所有$x$的个数的平方

解法:概率dp
第一阶段:
$cnt[i]$表示第一个生成的数字是$i$时还可以生成数字个数的期望,记$j$为下一个生成的数字,则cnt[0]为初始状态
$①i\le j$，执行$1$步骤,生成$j$,贡献个数为$cnt[j]+1$,则有$cnt[i]={\sum }_{j=i}^n{p}_j\ast (cnt[j]+1)$
$②i>j$,执行$3$步骤,生成$j$后结束,贡献个数为$1$,则有$cnt[i]={\sum }_{j=0}^{i-1}{p}_j\ast 1$
综上,$cnt[i]={\sum }_{j=0}^{i-1}{p}_j+{\sum }_{j=i}^n{p}_j(cnt[j]+1)={\sum }_{j=0}^{i-1}{p}_j+{\sum }_{j=i+1}^n{p}_j(cnt[j]+1)+p_i\ast (cnt[i]+1)$,移项化简可得$cnt[i]=$$\frac{{\sum }_{j=0}^i{p}_j+{\sum }_{j=i+1}^n{p}_j\ast (cnt[j]+1)}{(1-p_i)}$
第二阶段:
现在求 $\mathbb{E}(cn{t}^2)$,不过$\mathbb{E}(cn{t}^2)!={\mathbb{E}}^2(cnt)$,也得不到$\mathbb{D}(cnt)$,用$dp[i]$表示第一个选的数是$i$时的分数期望,则有$dp[i]=\mathbb{E}(cnt[i{]}^2)$,0为初始状态,答案为$dp[0]$
$①i>j$时,贡献个数为1,$1^2=1$,所以这部分的分数贡献期望为${\sum }_{j=0}^{i-1}{p}_j$
$②i\le j$时,贡献个数为$cnt[j]+1$,这部分的分数贡献期望为$p_j\ast \mathbb{E}((cnt[j]+1{)}^2)=p_j\ast (\mathbb{E}(cnt[j{]}^2)+2\ast \mathbb{E}(cnt)+1)$,即有$dp[i]={\sum }_{j=i}^n{p}_j\ast \mathbb{E}(cnt[j]+1{)}^2={\sum }_{j=i}^n{p}_j\ast (\mathbb{E}(cnt[j{]}^2)+2\ast \mathbb{E}(cnt)+1)={\sum }_{j=i}^n{p}_j\ast (dp[j]+2\ast cnt[j]+1)$
综上,$dp[i]={\sum }_{j=0}^{i-1}{p}_j+{\sum }_{j=i}^n{p}_j\ast (dp[j]+2\ast cnt[j]+1)$,移项化简得$dp[i]=$$\frac{{\sum }_{j=0}^i{p}_j+{\sum }_{j=i+1}^n{p}_j\ast (dp[j]+2\ast cnt[j]+1)+2\ast cnt[i]}{(1-p_i)}$

AcCode:

#include <iostream>
#include <algorithm>

#define int long long

const int mod = 998244353;
const int N = 110;
int p[N], cnt[N], sum[N], dp[N];

inline int quick_mow(int a, int b) {
	int ans = 1;
	a %= mod;
	while (b) {
		if (b & 1) ans = ans * a % mod;
		b >>= 1;
		a = a * a % mod;
	}
	return ans;
}

signed main() {
	int n, tsum = 0; scanf("%lld", &n);
	for (int i = 1; i <= n; i++) scanf("%lld", &p[i]), tsum += p[i];
	tsum = quick_mow(tsum, mod - 2);
	for (int i = 1; i <= n; i++) p[i] = p[i] * tsum % mod, sum[i] = (sum[i - 1] + p[i]) % mod;
	for (int i = n; i>=0; i--) {
		int temp = sum[i];
		for (int j = i + 1; j <= n; j++) {
			temp = (temp + p[j] * (cnt[j] + 1) % mod) % mod;
		}
		cnt[i] = temp * quick_mow((mod + 1 - p[i]) % mod, mod - 2) % mod;
	}
	for (int i = n; i >= 0; i--) {
		int temp = 2 * p[i] * cnt[i] % mod;
		for (int j = i + 1; j <= n; j++) {
			int tp = (dp[j] + 2 * cnt[j] + 1) % mod;
			temp = (temp + tp * p[j] % mod) % mod;
		}
		temp = (temp + sum[i]) % mod;
		dp[i] = temp * quick_mow((mod + 1 - p[i]) % mod, mod - 2) % mod;
	}
	printf("%lld\n", dp[0]);
}