生成函数强化训练

$\rm [集训队作业2013]城市规划$

考虑 $\rm EGF$ 的组合意义可以发现若设任意图方案数的 $\rm EGF$$F(x)$连通图的 $\rm EGF$$G(x)$显然任意图计数可以被看做先将 $n$ 个点分进若干个非空子集然后做连通图计数的方案数那么根据之前的结论

$$F(x) = \exp G(x) \iff G(x) = \ln F(x) $$

显然有 $F(x) = \sum _ n 2 ^ {\binom{n}{2}} x ^ n$于是只需直接对其求 $\ln$ 即可

时间复杂度 $\Theta(n \log n).$

$\rm Code~Link$

$\rm [CF438E]The~Child~and~Binary~Tree$

首先可以考虑一个 $\rm DP$$f_S$ 表示点权之和为 $S$ 的二叉树数量迭代时可以考虑枚举左子树的权值和 $S_l$右子树的权值和 $S_r$ 和根节点的权值 $w$那么有 $f_{S_l + S_r + w} = \sum _{S_l} \sum _{S_r} \sum _{w \in C} f_{S_l} f_{S_r}.$

容易发现这个式子可以用三个多项式的卷积来拟合$F(x) = \sum _ n f_n x ^ n, G(x) = \sum _ n [n \in C] x ^ n$于是有

$$F = F ^ 2 G + 1 $$

最后的 $+1$ 是为了补齐常数项因为 $f_0$$1.$

解方程可以得到

$$F = \frac{1 \pm \sqrt{1 - 4G}}{2G} $$

由于有两个解显然不可能都成立于是考虑 $x \to 0$ 时的特殊情况

$x \to 0$$F(x) \to 1,G(x) \to 0.$

$$\lim _ {x \to 0} \frac{1 + \sqrt{1 - 4G}}{2G} = +\infty \not= \lim _ {x \to 0} F $$
$$\lim _ {x \to 0} \frac{1 - \sqrt{1 - 4G}}{2G} = 1 = \lim _ {x \to 0} F $$

于是选取 $F= \frac{1 - \sqrt{1 - 4G}}{2G}$ 作为方程的解

到这一步我们发现一个问题因为 $G(x)$ 的常数项为 $0$所以 $G(x)$ 不可以求逆

考虑对解做变换这里选取分子有理化

$$F = \frac{\left(1 - \sqrt{1 - 4G}\right)\left(1 + \sqrt{1 - 4G}\right)}{2G\left(1 + \sqrt{1 - 4G}\right)} = \frac{2}{1 + \sqrt{1 - 4G}} $$

可以发现分母的常数项必定不为 $0$于是做多项式开方和多项式求逆即可

时间复杂度 $\Theta(n \log n).$

$\rm Code~Link$

$\rm [国家集训队]整数的~lqp~拆分$

$\{f_n\},\{g_n\}$ 分别为斐波那契数列和答案序列$F(x)$$G(x)$ 分别为它们的生成函数

显然有斐波那契数列的生成函数

$$F(x) = \frac{x}{1 - x - x ^ 2} $$

而对于 $G(x)$可以发现如果 $g_i$ 已经被求出来了那么给所有的拆分加上一个数 $n - i$ 即可得到 $g_n$ 的所有拆分$g_i$$g_n$ 的贡献为 $g_i \times f_{n - i}.$ 于是有 $g_n$ 的递推式

$$g_n = \sum _ {i = 0} ^ {n- 1} g_i \times f_{n - i} $$

$\{g_n\}$ 带入 $G(x)$ 可得

$$G(x) = \sum _ n x ^ n \sum _ {i = 0} ^ {n - 1} g_i f_{n - i} $$

可以发现等式右边实际上是卷积的形式于是有

$$G = GF $$

但是我们发现这样的话 $G(x) = 0$原因是 $g_0 = 0$于是我们强制 $g_0 = 1$得到

$$G = GF + 1 $$
$$G(x) = \frac{1}{1 - F(x)} = 1 + \frac{x}{1 - 2x - x ^ 2} $$

可以发现现在求出的 $G(x)$ 比实际值多了 $1$因为我们将 $g_0 = 0$ 强制变成了 $g_0 = 1$修正后有

$$G(x) = \frac{x}{1 - 2x - x ^ 2} = \frac{\frac{\sqrt{2}}{4}}{1 - (1 + \sqrt{2})x} + \frac{-\frac{\sqrt{2}}{4}}{1 - (1 - \sqrt{2})x} $$
$$[x ^ n]G(x) = \frac{\sqrt{2}}{4}(1 + \sqrt{2}) ^ n - \frac{\sqrt{2}}{4}(1 - \sqrt{2}) ^ n $$

最后求出 $\sqrt{2}$ 在模 $10 ^ 9 + 7$ 意义下的二次剩余并且用拓展欧拉定理即可计算出答案

时间复杂度 $\Theta(\log n).$

$\rm Code~Link$

$\rm [HAOI2018]染色$

首先考虑二项式反演

$f_i$ 表示恰好有 $i$ 个数的数量为 $S$ 的方案数$g_i$ 表示钦点 $i$ 个数的数量为 $S$ 的方案数显然有

$$g_k = \dbinom{m}{k} \times \frac{n ^ {\underline{kS}}}{(S!) ^ k} \times (m - k) ^ {n - kS} $$
$$g_k = \sum_{i = k} ^ m \dbinom{i}{k} f_i \iff f_k = \sum _ {i = k} ^ m (-1) ^ {i - k} \dbinom{i}{k} g_i $$
$$Ans = \sum _ {i = 0} ^ m w_if_i $$

于是我们考虑如何对于每个 $i \in [0, m]$ 求出 $f_i$这里有个技巧

$$\begin{aligned} f_k &= \sum _ {i = k} ^ m (-1) ^ {i - k} \frac{i!}{k!(i - k)!} g_i\\ &= k! \sum _ {i = k} ^ m i! g_i \times \frac{(-1) ^ {i - k}}{(i - k)!} \end{aligned} $$

容易发现上式可以用差值卷积计算时间复杂度 $\Theta(n \log n).$

$\rm 付公主的背包$

容易发现若设每种大小的物品的数量为 $f_i$那么答案的生成函数为

$$\prod _ i \left (\sum _ j x ^ {ji}\right) ^ {f _ i} = \prod _ i \left(\frac{1}{1 - x ^ i}\right) ^ {f _ i} $$

可以直接使用欧拉变换的求解方式

时间复杂度 $\Theta(n \log n).$

$无标号无根树计数$

考虑设 $f_n$ 表示大小为 $n$ 的无标号有根树的方案数$F(x)$ 为数列 $f$ 的生成函数

可以发现如果将一颗大小为 $n$ 的无标号有根树的根去除那么剩下的子树是一个个相同的子问题只要子树大小的和为 $n - 1$再加上根结点就可以唯一确定地拼出一颗无标号有根树于是有生成函数方程

$$F(x) = x \cdot \varepsilon \circ F(x) $$

求解这个生成函数方程有两种方法第一种是直接化简分治多项式乘法求解

$$F(x) = x \prod _ i (1 - x ^ i) ^ {-f_i} $$

考虑对两边取 $\ln$将连乘转连加

$$\ln F(x) = \ln x - \sum _ i f_i \ln (1 - x ^ i) $$

对数不好处理考虑求导

$$\frac{F'(x)}{F(x)} = \frac{1}{x} + \sum _ i if_i \times \frac{x ^ {i - 1}}{1 - x ^ i} $$

将两边同时乘以 $xF(x)$

$$xF'(x) = F(x) + F(x) \sum _ i if_i \frac{x ^ i}{1 - x ^ i} $$

考虑将右半部分还原成 $F(x)$ 表示

$$xF'(x) = F(x) + F(x)\left(\sum _ {i \geqslant 1} x ^ i F'(x ^ i)\right) $$

$G(x) = \sum _ k x ^ k F'(x ^ k)$简单推一推

$$G(x) = \sum _ k x ^ k \sum _ {i\geqslant 1} if _ i \left(x ^ {k}\right) ^ {i - 1} = \sum _ k \sum _ i i f _ i x ^ {ik} = \sum _ n x ^ n \sum _ {d | n} d f _ d $$

$g_n = \sum _ {d | n} df_d,g_1 = f_1 = 1.$

因此

$$f_n = \frac{1}{n - 1} \sum _ {k = 1} ^ {n - 1} f _ k g _ {n - k} $$

$f$ 使用分治多项式乘法求解$g$ 暴力求解即可做到 $\Theta(n \log ^ 2 n).$

另一种方法是使用牛顿迭代

显然我们要求解方程 $G \circ F(x) = F(x) - x \cdot \varepsilon \circ F(x) = 0$.

假设当前已经求出了方程在模 $x ^ n$ 意义下的解 $F_0(x)$设方程在模 $x ^ {2n}$ 意义下的解为 $F(x)$众所周知有

$$F(x) = F_0(x) - \frac{G \circ F_0(x)}{G' \circ F_0(x)} $$

我们知道$\varepsilon \circ F(x)$ 可以在 $\Theta(n \log n)$ 的时间复杂度内求出$F'(x)$ 可以 $\Theta(n)$所以可以在 $\Theta(n \log n)$ 的时间复杂度内求下式

$$F(x) = F_0(x) - \frac{F_0(x) - x \cdot \varepsilon \circ F_0(x)}{[F_0(x) - x \cdot \varepsilon \circ F_0(x)]'} $$

用上式迭代即可算出 $F(x)$时间复杂度 $\Theta(n \log n).$

现在我们已经求出了无标号有根树的方案数考虑将无标号无根树的方案数容斥出来

考虑钦点无标号无根树的根是它的重心于是只需要去掉根不是重心的无标号有根树的方案数可以了分类讨论

如果重心唯一那么一定存在一颗子树的大小大于 $\left\lfloor\frac{n}{2}\right\rfloor$考虑枚举它的大小 $i$容易发现这颗子树的方案数和将它切除后树的方案数都是无标号有根树计数问题其答案我们已经算出于是总方案数需要减去 $\sum _ {i = \left\lfloor\frac{n}{2}\right\rfloor + 1} ^{n - 1} f_i \times f_{n - i}.$

如果重心不唯一那么一棵树还会在两个重心上分别被计算这种方案只会在 $n$ 为偶数的情况下出现考虑到这两个重心一定相连于是将它们之间的连边断开后形成的两个子树的方案数是独立的但是我们发现当两颗子树完全相同时分别以它的两个重心为根时形成的有根树是同构的所以我们还是只会将它计算一次故算重的方案中不包括两颗子树相同的情况于是总方案数还需要减去 $\binom{f_{\frac{n}{2}}}{2}.$

综上问题得到解决时间复杂度为 $\Theta(n \log ^ 2 n)$$\Theta(n \log n).$

$\rm [CEOI2004]Sweets$

考虑构造 $F_i(x) = \sum _ {j = 0} ^ {m _ i} x ^ j$容易发现题目要求的就是 $\prod _ {i = 1} ^ n F_i(x)$ 的系数前缀和

于是再构造 $F_0(x) = \sum _ i x ^ i$将它和原本的 $n$ 个幂级数卷在一起现在考虑求 $F(x) = \prod _ {i = 0} ^ n F_i(x)$ 的第 $L$ 项系数

容易发现有

$$F(x) = \frac{(1 - x ^ {m_1 + 1})(1 - x ^ {m_2 + 1})\cdots(1 - x ^ {m_n + 1})}{(1 - x) ^ {n + 1}} $$

观察到 $n$ 很小考虑暴力将分子拆开于是分式变成了 $2 ^ n$ 个形如 $\frac{x ^ k}{(1 - x) ^ {n + 1}}$ 的部分之和

容易发现

$$[x ^ L]\frac{x ^ k}{(1 - x) ^ {n + 1}} = \dbinom{n + L - k}{n} $$

于是只需要求 $2 ^ {n + 1}$ 次形如 $\binom{t}{n}$ 的组合数即可

到这里我们又发现模数 $p$ 不是质数于是考虑将式子变形

$$\dbinom{t}{n}~\bmod~p = \frac{t^{\underline{n}}}{n!}~\bmod~p = \frac{t ^ {\underline{n}}~\bmod~n!\cdot p}{n!}~\bmod~p $$

于是这样就可以 $\Theta(n)$ 求解组合数了总时间复杂度 $\Theta(2^n n).$

$\rm [51nod1728]不动点$

简化题意 求有多少个从 $\{1,2,\cdots,n\}$$\{1,2,\cdots,n\}$ 的映射 $f$满足

$$\underbrace{f \circ f \circ \cdots \circ f}_{k}=\underbrace{f \circ f \circ \cdots \circ f}_{k-1} $$

保证 $nk \leqslant 2 \times 10 ^ 6,1\leqslant k \leqslant 3.$

可以发现这题本质上就是在求深度不超过 $k$环大小为 $1$ 的基环内向树森林的数量进一步发现其等价于树高不超过 $k$ 的有标号有根树森林的数量

考虑设树高不超过 $k$ 的有标号有根树数量的 $\rm EGF$$\hat{F}_k(x).$ 计算考虑递推深度不超过 $k$ 的树可以看作若干棵深度不超过 $k - 1$ 的树全部接在一个点上于是有

$$\hat{F}_k(x) = x\cdot \exp \hat{F}_{k - 1}(x) $$

考虑到需要求的是森林的数量于是答案的指数型生成函数为 $\exp \hat{F}_k(x).$

时间复杂度 $\Theta(kn\log n).$

$\rm [CF891E]Lust$

考虑一次对 $x$ 的操作造成的影响他会使 $a_x$ 减少 $1$答案增加 $\prod _ {i \not= x} a_i$$\prod _ i a_i$ 减少 $\prod _ {i \not= x} a_i.$

于是我们可以发现一次操作对答案的贡献等于 $\prod _ i a_i$ 的变化量进一步的最终答案等于 $k$ 次操作进行完后 $\prod _ i a_i$ 的变化量假设第 $a_i$ 被操作了 $b_i$那么答案为 $\prod _ i a_i - \prod _ i (a_i - b_i).$

考虑计算所有情况下 $\prod _ i (a_i - b_i)$ 的和最后再将答案除以 $n ^ k.$

假设有两个集合 $S,T,S \cap T = \varnothing$$f_i$ 表示 $\sum b_k = i,k\in S$ 对答案的贡献$g_i$ 表示 $\sum b_k = i,k\in T$ 对答案的贡献$h_i$ 表示 $\sum b_k = i,k\in S \cup T$ 对答案的贡献那么显然有

$$g_n = \sum _ {i = 0} ^ n \dbinom{n}{i} f_i \times g_{n - i} $$

因为 $f_i$$g_{n - i}$ 联合起来的贡献是它们的乘积并且由于操作有序所以将 $n$ 次操作分配到它们还导致要乘上 $\binom{n}{i}$ 的方案数

很明显这个式子可以用 $\rm EGF$ 来拟合$\hat{F}_i(x) = \sum _ j (a_i - j) \frac{x ^ j}{j!}$那么答案的生成函数为

$$\begin{aligned} \hat{F}(x) &= \prod _ i F_i(x)\\ &= \prod _ i \left( \sum _ j (a_i - j) \frac{x ^ j}{j!} \right)\\ &= \prod _ i \left( a_i \sum _ j \frac{x ^ j}{j!} - \sum _ {j} \frac{x ^ {j + 1}}{j!} \right)\\ &= \prod _ i (a_i - x)e ^ x\\ &= e ^ {nx} \prod _ i (a_i - x) \end{aligned} $$

$\prod _ i (a_i - x)$ 直接分治乘可以做到 $\Theta(n \log ^ 2 n)$$e ^ {nx}$ 的系数可以直接求由于 $\prod _ i (a_i - x)$ 的最高次数为 $n$所以直接枚举计算即可

总时间复杂度 $\Theta(n \log ^2 n).$