DTOJ 4848. 交通网络

题意

宫水三叶在玩一款游戏。

这是一款对城市进行建设的游戏，游戏里有 $n$ 个城市。三叶在任意两个城市上都修了一条普通公路，公路为双向边。

三叶并不满足于普通公路，于是三叶又把所有的连接 $(i,i+1)$ 的普通公路改造成了高速公路。

然而一次经济危机让三叶接近破产。为了减小公路的维修费，三叶决定拆除大部分公路。但是三叶还需要保证图联通，于是三叶决定保留原图的一棵生成树。

通过观察，三叶发现保留高速公路比保留普通公路更有价值，而且这些价值随着高速公路的增多会略快于指数级增加。

形象化的描述，如果三叶保留了 $x$ 条高速公路，那么新图的价值为 $x\cdot 2^x$ 。

虽然最优解很好求，但是三叶喜欢思考。她想知道对于原图的所有生成树，它们的价值的总和是多少。

本题采用捆绑测试。
对于所有数据，满足 $1\le n\le 5\times 10^5$ 。
每个测试点具体限制见下表。

测试点编号	$n$	分值
$1$	$\le 8$	$4$
$2$	$\le 18$	$6$
$3$	$\le 80$	$13$
$4$	$\le 500$	$14$
$5$	$\le 5000$	$15$
$6$	$\le 5\times 10^4$	$20$
$7$	$\le 5\times 10^5$	$28$

题解

由于$i$到$i+1$的边是特殊的，先考虑连了一些这种边，则剩下一些联通块。于是这些联通块的生成树个数问题就是经典的prufer计数 （并不会） ：$n^{m-2}{\prod }_{i=1}^{m}siz{e}_i$。由于不能保证剩下的关键边不被选，需要容斥一下算出恰好选$x$条边的方案数，DP可以做到$O(n^2)$。

考虑生成函数：每新增一个连通块的转移关系为乘上$f(x)={\sum }_{i=1}^{\infty }ni{x}^i$，至少$(n-i)$条关键边的方案即为$f^i(x)/n^2$，如果要直接容斥则要求出每一个$f^i(x)$不太现实。考虑构造一个式子使得恰好选$x$条关键边的方案的贡献为$x\ast 2^x$，注意到恰好选$x$条关键边在$f^{n-i}(x)$中记到的次数为$C_{x,i}$，如果求出${\sum }_{}^{}(n-i)f^i(x)$则恰好满足题意。对这个式子求和即可用多项式求逆算出。

再进一步考虑，把$f^i(x)$化为特征多项式，即可$O(n)$算出第$n$项的和。

这里需要用到广义二项式定理，即：$(1-x{)}^{-n}[x^m]=C((n+m-1),m)$，考虑$(1-x{)}^{-1}$的多项式为{$1,1,1,...$}，这个多项式的$n$次方考虑组合意义即可。