SRM 697 div1 hard

题目大意

有$n$个城市，每个城市有个权值$w_i$，任意两个城市$i,j$之间的道路数有$w_i*w_j$条。对于每种生成树，设每个点的度数为$d_i$，其权值定义为$\prod d_i$。问所有无根生成树的权值和。答案对$10^9+7$取模。
$n \leq 2000$

题解

因为与度数有关，所以很容易就能套上prufer序列。事实上，答案求的是这个东西：

$$\begin{aligned} &\sum_{a_1+\cdots+a_n=2n-2} \frac{(n-2)!}{(a_1-1)!\cdots(a_n-1)!}w_1^{a_1}\cdots w_n^{a_n}a_1\cdots a_n \\ =&w_1\cdots w_n\sum_{a_1+\cdots+a_n=n-2} \frac{(n-2)!}{a_1!\cdots a_n!}w_1^{a_1}\cdots w_n^{a_n}(a_1+1)\cdots (a_n+1)\\ =&w_1\cdots w_n(n-2)!\sum_{a_1+\cdots+a_n=n-2} \frac{1}{a_1!\cdots a_n!}w_1^{a_1}\cdots w_n^{a_n}(a_1+1)\cdots (a_n+1)\\ \end{aligned}$$
现在相当于要求
$$\sum_{a_1+\cdots+a_n=n-2} \frac{1}{a_1!\cdots a_n!}w_1^{a_1}\cdots w_n^{a_n}(a_1+1)\cdots (a_n+1)$$
我们把$(a_1+1)\cdots(a_n+1)$展开，单独考虑每个单项式的贡献，比如说考虑$a_1a_2a_3$，那么有
$$\begin{aligned} & \sum_{a_1+\cdots+a_n=n-2} \frac{1}{a_1!\cdots a_n!}w_1^{a_1}\cdots w_n^{a_n}a_1a_2a_3 \\ =&w_1w_2w_3\sum_{a_1+\cdots+a_n=n-2-3} \frac{1}{a_1!\cdots a_n!}w_1^{a_1}\cdots w_n^{a_n} \\ =&w_1w_2w_3\frac{(w_1+\cdots+w_n)^{n-2-3}}{(n-2-3)!} \end{aligned}$$
据此，我们可以推断出，原式就等于
$$\sum_{k=0}^{n-2} \left(\sum_{1\leq p_1 < p_2 <\cdots < p_k \leq n}\prod_{i=1}^k w_{p_i}\right) \frac{(w_1+\cdots+w_n)^{n-2-k}}{(n-2-k)!}$$
对于中间那一项，可以用一个$O(n^2)$ Dp直接求，那么整题就做完了。

总结：
1. prufer序列在含度数的生成树计数中异常有用
2. 把式子展开，考虑单项式的贡献