牛客2021多校第四场B（生成函数， 期望） 题解

成功用这道题生成函数入门了

题目描述

牛客2021多校第四场B题

生成$i\in [1,n]$​的数字的概率分别为$p_i$​​​

按照以下方式生成数字序列$a_i$

1. 生成数字$x$
2. 判断数字$x$​是否为$a_i$​​中的最大值, 如果是则重复1, 否则停止

通过以上方式生成一共$len$​个数, 则得分为$le{n}^2$​

求得分的期望

其中$p_i$以$\frac{w_i}{{\sum }_1^n{w}_i}$​的形式给出

输出概率对$998244353$取模
$$\begin{gathered} 0\le w_i\le 10^6 \\ 2\le n\le 100 \end{gathered}$$

题目分析

一个没有结束的数字序列一定是单调不降的, 比如$1,1,1,2,2,3$

如果一个序列的长度$len$​​​​​​​​​​超过$i$​​​​​​​​​​​时对应的生成概率为$p(len>i)$​​​​​​​​​​​​​​​​​,那么就可以知道这个序列下一个数停止的概率为$p(len>i)-p(len>i+1)$​​​​​​​​​​,那么再乘上$(i+1{)}^2$​​​​​​​就得到了$i+1$​​​​​​​​​​长度的得分期望

那么求$$\sum _0^{\infty }[p(len>i)-p(len>i+1)](i+1{)}^2$$​​​​就得到了总的期望

对这个式子进行一下化简
$$\begin{gathered} \sum _0^{\infty }[p(len>i)-p(len>i+1)](i+1{)}^2 \\ =\sum _0^{\infty }[p(len>i)(i+1{)}^2-p(len>i+1)(i+1{)}^2] \\ =p(len>0)(0+1{)}^2+\sum _1^{\infty }p(len>i)[(i+1{)}^2-i^2]-p(len>\infty )(\infty +1{)}^2 \\ =p(len>0)(0+1{)}^2+\sum _1^{\infty }p(len>i)[(i+1{)}^2-i^2] \\ =\sum _0^{\infty }p(len>i)[(i+1{)}^2-i^2] \\ =\sum _0^{\infty }p(len>i)(2i+1) \end{gathered}$$

那么如何求$p(len>i)$呢​​

构造一个生成函数$g_i(x)={\sum }_{k=0}^{\infty }{p}_i{x}^k=\frac{1}{1-p_{I}x}$的系数来表示对于第$i$个数生成了$k$次的生成概率,那么可以知道, 生成所有序列的生成函数,就是$f(x)={\prod }_{i=1}^n{g}_i(x)$ 其中$f(x)$的$x^i$的系数表示未结束的序列长度为$i$的概率, 因为生成的序列长度已经为$i$了,所以这个序列长度最终至少为$i+1$,因此这个系数就是$p(len>i)$​​​

因此$$f(x)=\sum _0^{\infty }p(len>i)x^i=\prod _{i=1}^n\frac{1}{1-p_{i}x}$$​​​

$$f^{\prime }(x)=\sum _0^{\infty }i\ast p(len>i)x^{i-1}=f(x)\ast (\sum _0^n\frac{p_i}{1-p_{i}x})$$​​​(复合函数的求导)

将$f(x)，f^{\prime }(x)$​代入原式
$$\begin{gathered} \sum _0^{\infty }p(len>i)(2i+1) \\ =2\ast \sum _0^{\infty }i\ast p(len>i)+\sum _0^{\infty }p(len>i) \\ =2\ast f^{\prime }(1)+f(1) \end{gathered}$$
这样就可以计算啦

复杂度是$O(n)$​

所以也许可以把n再扩大一些

题目踩坑

不管算多少步,每一步都要取模, 不要因为嫌麻烦在某一步不取模!

不知道什么时候就爆long long了

记得模意义下的除法是逆元,可以使用费马小定理

AC代码

代码丑陋，仅供参考

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int MAXN = 100;
const int MOD = 998244353;
typedef long long ll;
ll qp(ll a, int b)
{
    ll ans = 1;
    while(b)
    {
        if(b & 1)
        {
            ans = (ans * a) % MOD;
        }
        b >>= 1;
        a = (a * a) % MOD;
    }
    return ans;
}
ll inv(ll a)
{
    return qp(a, MOD - 2);
}
int wi[MAXN + 100];
int main()
{
    int n;
    ll sum = 0;
    scanf("%d", &n);
    for(int i = 1; i <= n; ++i)
    {
        scanf("%d", &wi[i]);
        sum += wi[i];
    }
    sum %= MOD;
    ll f1 = 1;
    ll df1 = 0;
    for(int i = 1; i <= n; ++i)
    {
        ll swi = ((sum - wi[i])%MOD + MOD) % MOD;
        f1 *= (sum * inv(swi)) % MOD;
        f1 %= MOD;
        df1 += (wi[i] *  inv(swi)) % MOD;
        df1 %= MOD;
    }
    df1 *= f1;
    df1 %= MOD;
    ll totans = ((df1 + df1) % MOD + f1) % MOD;
    printf("%lld", totans);
    return 0;
}