牛客多校day4-B.Sample Game

Sample Game

题意：

$1\sim n$个数，每个数都有一个随机概率$p_i$，进行如下操作：

1. 按照概率随机生成一个数。
2. 如果生成的数字不小于之前生成的任意一个数字，回到步骤$1$，否则到步骤$3$。
3. 如果生成数字的个数为$n$，那么贡献为$n^2$。

求期望贡献。

思路：

生成的序列肯定是一个非降的形式。我们假设最后数列的长度为$len$,那么可以得到长度大于$i$的概率$P(len>i)=\prod _{i=1}^n{p}_i^{t_i}$,其中$t_i$表示数字$i$出现的次数。因为在长度为$i$的时候，我们并不知道对于$i+1$的时候，取的数是会停下来还是会继续取，但是最终的长度一定会比$i$大，所以这里算的是$len>i$的概率。

那么我们可以得到$P(len=i)=P(len>i-1)-P(len>i)$。

所以我们贡献的期望就是:
$\begin{array}{rl}E(X)& =\sum _{i=1}^{\infty }{i}^{2}P(len=i)\\ & =\sum _{i=1}^{\infty }{i}^2(P(len>i-1)-P(len>i))\\ & =1^{2}P(len>0)+\sum _{i=1}^{\infty }((i+1{)}^2-i^2)P(len>i)\\ & =\sum _{i=0}^{\infty }(2i+1)P(len>i)\end{array}$

对于$P(len>i),i\in [0,+\infty )$，我们可以将这个数列写成生成函数的形式$f(x)=\sum _{i=0}^{\infty }P(len>i)x^i$。

可得$f^{{}^{\prime }}(x)=\sum _{i=0}^{\infty }iP(len>i)x^{i-1}$,
所以$E(x)=2f^{{}^{\prime }}(1)-f(1)$。

对于第$i$个数出现$j$次的概率，我们也可以写成一个生成函数$g_i(x)=\sum _{j=0}^{\infty }{p}_i^j{x}^j$。

根据生成函数中相乘的组合意义，我们又可以得到$f(x)=\prod _{i=1}^n{g}_i(x)$。

其中$g_i(x)$可以化简成$\frac{1}{1-p_{i}x}$的形式。
推导过程就是：
$\because g_i(x)=\sum _{j=0}^{\infty }{p}_i^j{x}^j$
$\therefore p_{i}x{g}_i(x)=\sum _{j=1}^{\infty }{p}_i^j{x}^j$
$\therefore (1-p_{i}x)g_i(x)=1$
$\therefore g_i(x)=\frac{1}{1-p_{i}x}$

所以可以得到:
$\begin{array}{rl}{f}^{{}^{\prime }}(x)& =\sum _{i=1}^n(g_i^{{}^{\prime }}(x)\prod _{j=1,j\ne i}^n{g}_j(x))\\ & =\sum _{i=1}^n(\frac{p_i}{(1-p_i{)}^2}\prod _{j=1,j\ne i}^n{g}_j(x))\\ & =\sum _{i=1}^n(g_i(x)\frac{p_i}{1-p_i}\prod _{j=1,j\ne i}^n{g}_j(x))\\ & =\sum _{i=1}^n(\frac{p_i}{1-p_i}\prod _{j=1}^n{g}_j(x))\\ & =f(x)\sum _{i=1}^n\frac{p_i}{1-p_i}\end{array}$

所以最后可以得到：$f(1)=\prod _{i=1}^n\frac{1}{1-p_i},f^{{}^{\prime }}(1)=f(1)\sum _{i=1}^n\frac{p_i}{1-p_i}$

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define int long long
#define mp make_pair
#define pb push_back
#define ls x<<1
#define rs x<<1|1
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define pii pair<int,int>
#define all(x) x.begin(),x.end()
#define cl(x,y) memset(x,y,sizeof(x))
#define nxtp(a,n) next_permutation(a+1,a+n+1)
#define mem(x,y,n) memset(x,y,sizeof(int)*(n+5))
const int N=1e6+10;
const int mod=998244353;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
const double INF=1e18;
using namespace std;
int p[N];
int qpow(int a,int b)
{
	int ans=1;
	while(b)
	{
		if(b&1)
			ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,i,sum=0;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		cin>>p[i];
		sum+=p[i];
	}
	sum=qpow(sum,mod-2);
	for(i=1;i<=n;i++)
		p[i]=p[i]*sum%mod;
	int f1=1,f2=0;
	for(i=1;i<=n;i++)
		f1=(f1*(1*qpow((1-p[i]+mod)%mod,mod-2)))%mod;
	for(i=1;i<=n;i++)
		f2=(f2+(p[i]*qpow((1-p[i]+mod)%mod,mod-2)))%mod;
	f2=2*f2*f1%mod;
	cout<<(f1+f2)%mod<<endl;
	return 0;
}