题意:
群里有 k k k个不同的复读机。为了庆祝平安夜的到来,在接下来的 n n n秒内,它们每秒钟都会选出一位优秀的复读机进行复读。非常滑稽的是,一个复读机只有总共复读了 d d d的倍数次才会感到快乐。问有多少种不同的安排方式使得所有的复读机都感到快乐 ( k ≤ 1000 , d ≤ 3 ) (k \le 1000, d \le 3) (k≤1000,d≤3)。
题解:
挺妙的,一个人的生成函数是 ∑ i = 0 ∞ [ d ∣ i ] i ! x i \sum_{i=0}^{\infty}\frac{[d|i]}{i!}x^i ∑i=0∞i![d∣i]xi。
这个 [ d ∣ i ] [d|i] [d∣i]跟FFT中那个idft挺像的,找个 d d d次单位根,变成 ∑ j = 0 d − 1 w d i j \sum_{j=0}^{d-1}w_{d}^{ij} ∑