2021牛客暑期多校训练营4-B题

B题：Sample Game

原题链接：https://ac.nowcoder.com/acm/contest/11255/B

题目大意

其中$p_i=\frac{w_i}{{\sum }_{j=x+1}^n}$
答案对 998244353取模

思路

对于x 令$f_x=E(x)$
期望值为
$$f_x=\sum _{i=1}^{x-1}{p}_i+p_x\cdot (1+f_x)+\sum _{i=x+1}^n{p}_i(f_i+1)$$
$$=>f_x=1+p_x\cdot f_x+\sum _{i=x+1}^n{p}_i\cdot f_i$$
化简得
$$f_x=\frac{1+{\sum }_{i=x+1}^n{p}_i\cdot f_i}{1-p_x}$$
从n->1推导即可。
对x² 因为$E(x^2)\ne E^2(x)$
根据期望的的性质，$E((x+1{)}^2)=E(x^2+2x+1)=E(x^2)+2E(x)+E(1)$
令$g_x=E(x^2)$
$$g_x=1\cdot \sum _{i=1}^x{p}_i+p_x\cdot (g_x+2f_x+1)+\sum _{i=x+1}^n{p}_i(g_x+2f_x+1)$$
$$=>g_x=\frac{1+2p_x{f}_x+{\sum }_{i=x+1}^n{p}_i(g_i+2f_i)}{1-p_x}$$
同理n->1推导即可。
最后
$$Ans=\sum _{i=1}^n\cdot p_i\cdot (g_i+2f_i+1)$$
结果可能很大，所以要记得取模。还需要用到费马小定理和快速幂逆元。

代码实现

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod=998244353;
ll w[110],f[110],g[110],s=0,p[110];
ll fmx(ll n)
{
	ll a=1,k=mod-2;
	while(k)
	{
		if(k&1)
		a=a*n%mod;
		n=n*n%mod;
		k>>=1;
	}
	return a;
}
ll fun(int x,int n)
{
	ll sum=0;
	for(int i=x;i<=n;i++){
		sum+=w[i]%mod*f[i]%mod;
		sum%=mod;
	}
	return sum;
}
ll gg(int x,int n)
{
	ll sum=0;
	for(int i=x;i<=n;i++){
		sum+=w[i]*(g[i]+2*f[i])%mod;
		sum%=mod;
	}
	return sum;
}
int main()
{
	int n;
	ll ans=0;
	cin>>n;
	for(int i=1;i<=n;i++)
	cin>>w[i],s+=w[i];
	for(int i=1;i<=n;i++)
	p[i]=w[i]*fmx(s)%mod;
	for(int i=n;i>=1;i--)
		f[i]=(s+fun(i+1,n))%mod*fmx(s-w[i])%mod;
	for(int i=n;i>=1;i--)
		g[i]=(s+2*w[i]*f[i]+gg(i+1,n))%mod*fmx(s-w[i])%mod;
		
	for(int i=1;i<=n;i++)
		ans+=p[i]*(g[i]+2*f[i])%mod;
	cout<<(ans+1)%mod;
	return 0;
}