[BZOJ3456] 城市规划（的拓展）

前言

最近看到一道经典的题目，就是这一题了。

题意：求 $n$ 个点的无向连通图的个数。

题意很简洁，做法也很自然。因为本篇重点不在这一道题，因此这里只做一个简短的说明：

不妨设答案的生成函数为 $\sum _{i=1}^{\infty }{f}_i{x}^i$ 。

我们知道我们可以取任意多任意大小的连通块生成一张无向图，而 $n$ 个点的无向图个数共有 $2^{\frac{n(n-1)}{2}}$ 个。

设 $f(x)=\sum _{i=1}^{\infty }\frac{f_i}{i!}{x}^i$ 。

根据指数型生成函数的知识我们有：$$e^{f(x)}=\sum _{i=0}^{\infty }\frac{2^{\frac{i(i-1)}{2}}}{i!}{x}^i$$

多项式求 $\ln$ 即可。

正文

显然，联通图可以理解为连通块个数等于 $1$ 。

如果是连通块个数等于 $k$ 的方案数，我们只需要求 $f^k$ 就可以了，注意到同一张图会被生成 $k!$ 次（生成的排列共有 $k!$ 种），因此答案的指数型生成函数是 $\frac{f^k}{k!}$ 。

那如果是连通块个数 $\le k$ 的方案数呢？

很显然，把 $1$ 至 $k$ 的函数求和即可（下文为了方便实际上还把函数加了一个常数 $1$ ，不过并没有什么影响）：

$$G(x)=e^x\mathrm{m}\mathrm{o}\mathrm{d}{x}^{k+1}$$

$$F(x)=G(f(x))=\sum _{i=0}^k\frac{f^i(x)}{i!}$$

现在的问题在于如何求 $F(x)$。

暴力乘卷积优化的复杂度为 $O(n^2\log n)$ 。

使用快速计算复合函数的算法的话，就是 $O((n\log n{)}^{1.5})$ 。

当然，以上两种做法都没有用到 $F$ 的特殊性质。

我们知道 $e^x$ 的重要特性是导数与原函数相等。

而 $$G(f)=e^f\mathrm{m}\mathrm{o}\mathrm{d}{f}^{k+1}$$

不妨对两边同时求导：

$$G^{\prime }(f)=e^f\mathrm{m}\mathrm{o}\mathrm{d}{f}^k$$

两式对减：

$$G(f)-G^{\prime }(f)=\frac{f^k}{k!}$$

用 $F$ 替换 $G$ ：

$$F(x)-\frac{F^{\prime }(x)}{f^{\prime }(x)}=\frac{f^k(x)}{k!}$$

这就是一个很典型的一阶微分方程了。

于是我们有：

$$F^{\prime }(x)-F(x)f^{\prime }(x)=-\frac{f^k(x)f^{\prime }(x)}{k!}$$

$$e^{-f(x)}(F^{\prime }(x)-F(x)f^{\prime }(x))=-\frac{f^k(x)f^{\prime }(x)e^{-f(x)}}{k!}$$

$$(F(x)e^{-f(x)}{)}^{\prime }=-\frac{f^k(x)f^{\prime }(x)e^{-f(x)}}{k!}$$

因此有：

$$F(x)=e^{f(x)}\int -\frac{f^k(x)f^{\prime }(x)e^{-f(x)}}{k!}dx$$

这样子就可以在 $O(n\log n)$ 时间内求解了。

个人认为这个解题思路相当巧妙，可以优化许多类似问题。