ABC231G Balls in Boxes

刚开始推的时候第一步就忘了乘方案数...

不妨将答案的式子列出:

\[\frac{1}{n^k}\sum_{\sum d_i=k}\frac{k!}{\prod{d_i!}}\prod_{1 \le i \le n}(A_i+d_i) \]

\[\frac{k!}{n^k}\sum_{\sum d_i=k}\prod_{1 \le i \le n}\frac{(A_i+d_i)}{d_i!} \]

这种形式使我们自然联想到 EGF,对于第 \(i\) 个元素有:

\[\begin{aligned} F_i(x)&=\sum_{j \ge 0}\frac{A_i+j}{j!}x^j \\ &=A_ie^x+xe^x \end{aligned}\]

那么:

\[\prod_{1 \le i \le n}F_i(x)=e^{nx}\prod_{1\le i\le n}(A_i+x) \]

注意到后面次数为 \(n\) ,可以通过分治fft或者暴力计算系数

前面 \(e^{nx}=\sum_{i \ge 0}\frac{n^i}{i!}x^i\)\(k-i\) 很大直接计算肯定不行

但是我们可以将它和最前面的 \(k!\) 合并成一个下降幂,预处理即可。

时间复杂度为 \(\mathcal O(n\log^2n)\)\(\mathcal O(n^2)\)

posted @ 2022-04-10 21:41  chihik  阅读(11)  评论(0编辑  收藏  举报