CF622F The Sum of the k-th Powers 题解
观前提示
本题解仅提供一个理论复杂度正确的解法,因为本题模数为 \(10^9+7\),没有优秀 \(\text{MTT}\) 板子的我被卡常了。
正文部分
不妨设 \(S_{n,m}=\sum_{i=0}^{n-1}i^m\),答案就是 \(S_{n+1,k}\)。
再设:
\[
\begin{aligned}
F(x) &= \sum_{i \geq 0} f_i x^i \\
&= \sum_{i \geq 0} \frac{S_{n,m}x^i}{i!} \\
&= \sum_{i \geq 0} \sum_{j=0}^{n-1} \frac{(jx)^i}{i!} \\
&= \sum_{j=0}^{n-1} \sum_{i \geq 0} \frac{(jx)^i}{i!} \\
&= \sum_{i=0}^{n-1}e^{ix} \\
&= \frac{e^{nx}-1}{e^x-1} \\
&= \frac{x}{e^x-1} \cdot \frac{e^{nx}-1}{x}
\end{aligned}
\]
继续设:
\[
\begin{aligned}
G(x) &= \sum_{i \geq 0} \frac{g_ix^i}{i!}=\frac{x}{e^x-1} \\
F(x) &= G(x) \cdot \frac{e^{nx}-1}{x} \\
&= (\sum_{i \geq 0} \frac{g_ix^i}{i!}) \cdot (\sum_{i \geq 1}\frac{n^ix^{i-1}}{i!}) \\
&= (\sum_{i \geq 0} \frac{g_ix^i}{i!}) \cdot (\sum_{i \geq 0}\frac{n^{i+1}x^i}{(i+1)!}) \\
S_{n,m} &= m!f_m \\
&= m! \sum_{i=0}^m \frac{g_i}{i!} \cdot \frac{n^{m-i+1}}{(m-i+1)!} \\
&= \frac{1}{m+1}\sum_{i=0}^m \binom{m+1}{i}g_in^{m-i+1}
\end{aligned}
\]
又有 \(\large G(x)=\frac{x}{e^x-1}=(\frac{e^x-1}{x})^{-1}=(\sum_{i \geq 0}\frac{x^i}{(i+1)!})^{-1}\),用 \(\text{MTT}\) 求逆即可在 \(\mathcal{O}(k \log k)\) 的时间复杂度内得到 \(g_i\),进而求出 \(S_{n+1,k}\)。
关于 \(G(x)\)
把 \(g_i\) 求出来后,一个不搞 \(\text{OI}\) 的同学一眼认出这个数列就是伯努利数。
所以 \(G(x)\) 即为伯努利数的 \(\text{EGF}\),上面的解法也是 \(\mathcal{O}(n \log n)\) 求伯努利数的方法。