题解 [WC2019] 数树

题意

对于两棵树 $T_1,T_2$ （编号都是 $1\sim n$），我们给每个点染一个 $[1,y]$ 之间的一个整数，如果两个点 $p,q$ 之间的路径序列在两棵树上都一样，那么它们必须被赋予相同颜色。不妨设其为 $w(T_1,T_2)$。

问题 0：给定 $T_1,T_2$，求 $w(T_1,T_2)$。

问题 1：给定 $T_1$，求 ${\sum }_{T_2}w(T_1,T_2)$。

问题 2：求 ${\sum }_{T_1,T_2}w(T_1,T_2)$。

答案对 $998244353$ 取模。

数据范围：$3\le n\le 10^5,1\le y<998244353$。

问题 0

稍微看看就可以发现问题等价于
$$y^{n-|E_1\cap E_2|}$$
其中 $E_1\cap E_2$ 是边的交。

问题 1

$$y^n\sum _{T_2}{y}^{-|E_1\cap E_2|}=y^n\sum _S{y}^{-|S|}\sum _{T_2}[S=E_1\cap E_2]$$

令 $F(S)={\sum }_{T_2}[S=E_1\cap E_2]$，我们考虑作出 $G(S)={\sum }_{T_2}[S\subseteq E_2]$，则
$$G(S)=\sum _{S\subseteq T}F(T)$$
使用容斥${}^{[1]}$得到
$$F(S)=\sum _{S\subseteq T}(-1{)}^{|T|-|S|}G(T)$$
于是
$$\begin{gathered} y^n\sum _{S\subseteq T_1}{y}^{-|S|}\sum _{S\subseteq T}(-1{)}^{|T|-|S|}G(T) \\ =y^n\sum _{T\subseteq T_1}(-1{)}^{|T|}G(T)\sum _{S\subseteq T}(-y{)}^{-|S|} \\ =y^n\sum _{T\subseteq T_1}(-1{)}^{|T|}G(T)\sum _k\left(\genfrac{}{}{0ex}{}{|T|}{k}\right)(-\frac{1}{y}{)}^k \\ =y^n\sum _{T\subseteq T_1}(\frac{1}{y}-1{)}^{|T|}G(T) \end{gathered}$$
一个基于 prufer 序的知名结论${}^{[2]}$给出
$$G(T)=n^{k-2}\prod _{i=1}^k{a}_i$$
于是原式化为
$$\begin{gathered} \frac{y^n}{n^2}\sum _{T\subseteq T_1}(\frac{1}{y}-1{)}^{|T|}\prod _{i=1}^k(n{a}_i \\ =\frac{y^n}{n^2}\sum _{T\subseteq T_1}(\frac{1}{y}-1{)}^{n-k}\prod _{i=1}^{k}n{a}_i \\ =\frac{(1-y{)}^n}{n^2}\sum _{T\subseteq T_1}\prod _{i=1}^k(\frac{y}{1-y}n{a}_i) \end{gathered}$$
我们使用树上背包来解决这个问题，设 $dp(u,i)$ 表示考虑 $u$ 的子树，$u$ 所在连通块大小为 $i$，不算上当前连通块的贡献，一开始 $f(u,1)=1$，其余的都是 0，则每次转移：
$$dp(u,i)\leftarrow f(u,i)\sum _j\frac{yn}{1-y}jf(v,j)+\sum _{j}f(v,j)f(u,i-j)$$
做出每个点的 dp 生成函数，那么
$$F_u(x)\leftarrow F_u(x)\frac{yn}{1-y}{F}_v^{\prime }(1)+F_u(x)F_v(x)$$
记录每个生成函数的开销太大，但我们考虑最后我们问的是 $\frac{yn}{1-y}{F}_v^{\prime }(1)$，所以我们不妨考虑每个点上这个值的变化：
$$F_u^{\prime }(1)\leftarrow \frac{yn}{1-y}{F}_v^{\prime }(1)F_u^{\prime }(1)+F_u(1)F_v^{\prime }(1)+F_v(1)F_u^{\prime }(1)$$
记录每个点的 $F_u(1),F_u^{\prime }(1)$，那么总体可以 $O(n)$ 树上 dp 将其求出。

问题 2

$$\begin{gathered} y^n\sum _{T_1}\sum _{T\subseteq T_1}(\frac{1}{y}-1{)}^{|T|}G(T) \\ =y^n\sum _T(\frac{1}{y}-1{)}^{|T|}G(T{)}^2 \\ =y^n\sum _T(\frac{1}{y}-1{)}^{|T|}{n}^{2k-4}\prod _{i=1}^k{a}_i^2 \\ =\frac{(1-y{)}^n}{n^4}\sum _T\prod _{i=1}^k(\frac{y}{1-y})(n{a}_i{)}^2 \end{gathered}$$

设一个 $T$ 的生成函数
$$F(x)=(\frac{y}{1-y})n^2\sum _{i\ge 1}\frac{i^i}{i!}{x}^i$$
则答案是
$$\frac{(1-y{)}^n}{n^4}[x^n]n!\exp F(x)$$
一遍 exp 即可求出，复杂度 $O(n\log n)$。

注释

[1] 容斥

结论

已知全集 $\Omega$，若对所有子集 $S$ 都有

$$G(S)=\sum _{S\subseteq T\subseteq \Omega }F(T)$$

则对所有子集 $T$ 都有

$$F(S)=\sum _{S\subseteq T\subseteq \Omega }(-1{)}^{|T|-|S|}G(T)$$

证明

$$\begin{gathered} \sum _{S\subseteq T\subseteq \Omega }(-1{)}^{|T|-|S|}G(T)=\sum _{S\subseteq K\subseteq \Omega }F(K)\sum _{S\subseteq T\subseteq K}(-1{)}^{|T|-|S|} \\ =\sum _{S\subseteq K\subseteq \Omega }F(K)\sum _{k\ge 0}\left(\genfrac{}{}{0ex}{}{|K|-|S|}{k}\right)(-1{)}^k=\sum _{S\subseteq K\subseteq \Omega }F(K)[K=S] \\ =F(S) \end{gathered}$$

[2] 知名结论

结论

点有标号，将有 $k$ 个连通块（大小分别是 $a_1,a_2,\cdots ,a_k,\sum a_i=n$）的森林连成一个树的方案数为

$$n^{k-2}\prod _{i=1}^k{a}_i$$

证明

第一步转化：考虑一个大小为 $k$ 的有标号树，设第 $i$ 个点的度为 $d_i$，让每个点代表原来的一个连通块，则这 $d_i$ 个度任意分配，所以每个这样的树都对答案贡献

$$\prod _{i=1}^k{a}_i^{d_i}$$

第二步转化：考虑 prufer 序与树一一对应，而数组是 prufer 序的充要条件是：大小为 $k-2$，值域为 $[1,k]$。且我们知道 $d_i=c_i+1$，$c_i$ 是 $i$ 在 prufer 序中出现的次数。所以每个这样的 prufer 序都对答案贡献：

$$\prod _{i=1}^k{a}_i^{c_i+1}$$

现在设出每个数的 EGF：

$$\sum _{j\ge 0}\frac{a_i^j}{j!}{x}^j=e^{a_{i}x}$$

那么答案是

$$(k-2)!\prod _{i=1}^k{a}_i[x^{k-2}]\prod _{i=1}^k{e}^{a_{i}x}=n^{k-2}\prod _{i=1}^k{a}_i$$