【LOJ2320】「清华集训 2017」生成树计数

【题目链接】

• 点击打开链接
  

【思路要点】

• 连上\(a_i\)的限制，题目要求的实际上是\(\sum_{T}\prod_{i=1}^{N}a_i^{d_i}*d_i^{M}\sum_{i=1}^{N}d_i^{M}\)。
• 我们知道树的Prufer序列与树点的度数密切相关，因此考虑使用Prufer序列解题。
• Prufer序列提供了一个长度为\(N-2\)，各个元素在\(N\)以内的正整数序列和\(N\)个节点的有标号无根树的双射，每个元素在Prufer序列中出现的次数+1就是其在树上的点度。
• 考虑给定式子的组合意义：写出一个Prufer序列，在序列后补足1到\(N\)各一个，使得各数出现次数等于其树上的点度，在每一种数中选择一个位置涂色（每次涂的颜色不同，且不会覆盖），重复\(M\)次；再选取一个数\(X\)，重复对某个写有\(X\)的位置涂色\(M\)次，最后将方案数乘以这个数列上各个数权值（\(a_i\)）的乘积，将结果加入答案。
• 记\(f_{i,j}\)表示考虑了1到\(i\)，尚未选取过\(X\)，Prufer序列中的\(N-2\)个位置有\(j\)个位置已经涂了色，可行的方案数与已经考虑过的数的权值乘积的和；\(g_{i,j}\)表示考虑了1到\(i\)，已经选取过\(X\)，Prufer序列中的\(N-2\)个位置有\(j\)个位置已经涂了色，可行的方案数与已经考虑过的数的权值乘积的和。
• 转移时枚举对于\(i\)，有多少Prufer序列中的位置被涂了色，则有$$f_{i,j}=\sum_{k=0}^{M}f_{i-1,j-k}*\binom{N-2-j+k}{k}*a_i^{k+1}*(S(M,k)*k!+S(M,k+1)*(k+1)!)$$ $$g_{i,j}=\sum_{k=0}^{2*M}f_{i-1,j-k}*\binom{N-2-j+k}{k}*a_i^{k+1}*(S(2*M,k)*k!+S(2*M,k+1)*(k+1)!)$$ $$g_{i,j}=g_{i,j}+\sum_{k=0}^{M}g_{i-1,j-k}*\binom{N-2-j+k}{k}*a_i^{k+1}*(S(M,k)*k!+S(M,k+1)*(k+1)!)$$
• 其中\(S(M,k)\)表示第二类斯特林数。