2022 CCPC桂林站 D Alice‘s Dolls 题解

题目链接：https://codeforces.com/gym/104008/problem/D
题意：抽一次有 $p$ 的概率抽到东西，$(1-p)$ 的概率抽不到，对于所有的 $1\le k\le m$ 求出抽到第 $n$ 次所用次数的 $k$ 次方的期望。
$1\le m\le 10^5,1\le n\le 10^5,p=\frac{a}{b}$

令 $X_i$ 表示已经抽到了 $i$ 个时，再抽到一次时所抽的次数。$Y$ 表示抽到 $n$ 个时抽的次数。那么显然有 $Y={\sum }_{i=0}^{n-1}{X}_i$。

$Y^k=({\sum }_{i=0}^{n-1}{X}_i{)}^k$, 其在期望意义下也成立，即 $E{Y}^k=E(({\sum }_{i=0}^{n-1}{X}_i{)}^k)$，暴力展开：
$$\begin{array}{rl}E{Y}^k& =\sum _{a_0+a_2+...+a_{n-1}=k}\left(\genfrac{}{}{0ex}{}{k}{a_0,a_1,...,a_{n-1}}\right)\prod _{i=0}^{n-1}E(X_i^{a_i})\\ \frac{E{Y}^k}{k!}& =\sum _{a_0+a_2+...+a_{n-1}=k}\prod _{i=0}^{n-1}\frac{E(X_i^{a_i})}{a_i!}(X的下标不同的项独立，因此可以拆开期望EXY=EXEY)\end{array}$$
同时可以发现 $X_i$ 之间独立同分布，因此 $E{X}_i$ 两两都是相等的，不妨都变成 $X$。

只需要设 E.G.F. $F(x)={\sum }_{i\ge 0}\frac{E{X}^i}{i!}{x}^i$。于是 $E{Y}^k=[x^k]F(x{)}^n$

现在的问题变为求 $E{X}^k$。

容易发现:
$$\begin{gathered} X=\{\begin{array}{ll}1& 直接抽到\\ 1+X& 没有抽到\end{array} \\ {X}^k=\{\begin{array}{ll}1& 直接抽到\\ (1+X{)}^k& 没有抽到\end{array} \end{gathered}$$

根据条件期望公式：
$$\begin{gathered} E{X}^k=p+(1-p)E((1+X{)}^k) \\ =1+\frac{1-p}{p}\sum _{i=0}^{k-1}E{X}^i(\genfrac{}{}{0ex}{}{k}{i}) \\ \frac{E{X}^k}{k!}=1+\frac{1-p}{p}\sum _{i=0}^{k-1}\frac{E{X}^i}{i!}\frac{1}{(k-i)!} \end{gathered}$$
令 $G[k]=$$\frac{E{X}^k}{k!}$，可以发现 $G[k]$ 只与 $G[0]\sim G[k-1]$ 有关，可以分支FFT得到所有的 $G[i]$。