2021牛客多校#4 B- Sample Game

题目链接

传送门

题目大意

有一个可以生成$1$到$n(1\le n\le 100)$的随机数生成器，生成$x(1\le x\le n)$的概率为$p_x$。
输入$W_i$，$p_i$的计算公式为$\frac{w_i}{{\sum }_{j=1}^n{W}_j}$.
现在一次执行以下操作:
1.随机生成一个数$x$；
2.若$x$是之前生成的数中最大的，则执行步骤1，否则执行步骤3；
3.若共生成了$y$个数字，则得到分数$y^2$；
求获得分数的期望值。

题解

易得$x$的期望为$$f_x=1\times \sum _{i=1}^{x-1}{p}_i+p_x(1+f_x)+\sum _{i=x+1}^n{p}_i(1+f_i)$$
经化简得$$f_x=\frac{1+\sum _{i=x+1}^n{p}_i{f}_i}{1-p_x}$$
由于$f(x^2)!=f(x{)}^2$,所以无法直接得到结果，但$f(x+1{)}^2=f(x{)}^2+2f(x)+1$，所以我们可以用一个新的$dp$数组来储存$f(x{)}^2$.
易得$$d{p}_x=1\times \sum _{i=1}^{x-1}{p}_i+p_x(1+2\times f_x+d{p}_x)+\sum _{i=x+1}^n{p}_i(1+2\times f_i+d{p}_i)$$
经化简得$$d{p}_x=\frac{1+2p_x{f}_x+\sum _{i=x+1}^n{p}_i(d{p}_i+2\times f_i)}{1-p_x}$$
然后统计其中$d{p}_i+2\ast f_i+1$的总值
经化简得$$ans=1+\sum _{i=1}^n{p}_i(d{p}_i+2\times f_i);$$
注意结果全程要取余$mod=998244353$，还要采用费马小定理即快速幂来取余（超范围的全给他模上【狗头】）。

参考代码

#include<bits/stdc++.h>
#define ll long long
const ll mod=998244353;
const int N=1e2+5;
ll dp[N],f[N],w[N],p[N],sum[N],num[N];
ll n,s;
ll pow(ll a,ll b)
{
    ll ret=1;
    while(b)
    {
        if(b&1)
            ret=ret*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ret;
}
int main()
{
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&w[i]);
        s+=w[i];
    }
    for(int i=1;i<=n;i++)
        p[i]=pow(s,mod-2)*w[i]%mod;
    num[n]=0;
    for(int i=n;i>=1;i--)
    {
        f[i]=(1+num[i])*pow((1+mod-p[i]),mod-2)%mod;
        num[i-1]=(f[i]*p[i]%mod+num[i])%mod;
    }
    num[n]=0;
    for(int i=n;i>=1;i--)
    {
        dp[i]=(1+2*f[i]*p[i]%mod+num[i])*pow((1+mod-p[i]),mod-2)%mod;
        num[i-1]=(p[i]*(dp[i]+2*f[i])+num[i])%mod;
    }
    printf("%lld\n",(num[0]+1)%mod);
}