【湖南集训 4.1】巧克力

题目描述

你有$C$种不同的巧克力，每种都有无限个。
问有次序地取出$n$个后，个数为奇数的巧克力种类数恰为$m$的概率。

$n\leq 10^9, 0\leq m \leq C\leq 10^5$

---

分析

考虑生成函数。令

$$f(x)=\left (\frac{1}{0!}+\frac{x^2}{2!}+\frac{x^4}{4!}\cdots \right)^{C-m}\left (\frac{x}{1!}+\frac{x^3}{3!}+\frac{x^5}{5!}\cdots \right )^m$$

那么答案就是

$$\binom{C}{m}\frac{n!_{[x^n]}f(x)}{C^n}$$

上面那两个式子的生成函数，其实分别是

$$\frac{e^x-e^{-x}}{2}$$

和

$$\frac{e^x+e^{-x}}{2}$$

（这部分考场上没推出来，一直纠结怎么把奇数项和偶数项提出来，实际上这么简单的小技巧是要记住的。

那么剩下的就顺理成章了。

$$f(x)=\left(\frac{e^x-e^{-x}}{2}\right)^{C-m}\left (\frac{e^x+e^{-x}}{2} \right )^m$$
$$=\left(\frac{e^{2x}-1}{2e^x}\right)^{C-m}\left (\frac{e^{2x}+1}{2e^x} \right )^m$$
$$=\frac{\left(e^{2x}-1\right)^{C-m}\left (e^{2x}+1 \right )^m}{2^Ce^{Cx}}$$

换元$y=e^{2x}$

$$f(x)=\frac{\left(y-1\right)^{C-m}\left (y+1 \right )^m}{2^Ce^{Cx}}$$

现在要求出这个函数的$n$次项系数。

想一下这个东西乘开以后长啥样，大概是一个二次项展开的形式。

$$f(x)=\frac{\sum_{i=0}^{max(m, C-m)} d_iy^i}{2^Ce^{Cx}}$$

这里下面除的一个$e^{Cx}$实际上可以直接在对应的系数上减掉。
用快速幂FFT可以$O(ClogC)$算出$d_i$

把$y$带回去，我们要考虑$e^{2ix}$对$n$次项的贡献。
实际上$e^{2ix}$泰勒展开以后长这个样

$$\frac{1}{0!}+\frac{(2ix)^{1}}{1!}+\frac{(2ix)^2}{2!}\cdots$$

因此，它对$n$次项的贡献显然就是

$$\frac{(2i)^n}{n!}$$

而我们最后乘上的$n!$恰好与这里的抵消掉了。

于是问题就这样被解决了。

时间复杂度$O(ClogC)$
空间复杂度$O(C)$