【UOJ#450】[集训队作业2018] 复读机

题目链接

题目描述

群里有$k$个不同的复读机。为了庆祝平安夜的到来，在接下来的$n$秒内，它们每秒钟都会选出一位优秀的复读机进行复读。非常滑稽的是，一个复读机只有总共复读了$d$的倍数次才会感到快乐。问有多少种不同的安排方式使得所有的复读机都感到快乐。

Sol

发现 $d$ 只有 $3$ ， 很可能需要分开讨论。

$d=1$ 就是 $k^n$

$d=2$:
其实容易发现这是一个有次数限制的可重排列问题，那么可以使用指数型生成函数来解决。
一个复读机的生成函数就是:
$$G(x)=\sum _{i=0}\frac{x^{2i}}{(2i)!}$$
其实就是去除掉所有奇数项，那么这是一个常见的套路了，就是:
$$G(x)=\frac{e^x+e^{-x}}{2}$$
泰勒展开后作差就是原来的式子了。

我们要求的东西就是:
$$n!\ast G(x{)}^k[x^n]$$
这里直接二项式定理展开:
$$n!\left(\frac{(e^x+e^{-x}{)}^k}{2^k}\right)=n!\left(\frac{{\sum }_{i=0}^k{e}^{(2i-k)x}\left(\genfrac{}{}{0ex}{}{k}{i}\right)}{2^k}\right)$$

里面泰勒逆展开后发现 $[x^n]$ 下自带一个