这类问题一般是如下形式:
给定 n , k n,k n,k,求:
∑ i = 1 n i k \sum_{i=1}^ni^k i=1∑nik
如果在模数为质数的情况下,可以用拉格朗日差值的方法,在 O ( k l o g k ) O(klogk) O(klogk)的时间内算出来。
如果有时不取模或者可以用浮点数,或者有其他用途时,可以用高斯消元在 k 3 k^3 k3的时间内处理,可以参考这道题。
如果是 N T T NTT NTT模数,或者可以用浮点数的话,也可以使用伯努利数和FFT来求,可以参考这里,复杂度 O ( k l o g k ) O(klogk) O(klogk)
但是有时取模,模数不是质数,那么就不好求取逆元了,所以我们还可以用第二类斯特林数来求,根据公式(假设 S n k S_{n}^k Snk为第二类斯特林数, C n m C_n^m Cnm为组合数):
x k = ∑ i = 0 k S i k x i = ∑ i = 0 k S i k C x i i ! \begin{aligned} x^k=&\sum_{i=0}^kS_{i}^kx^i \\ =&\sum_{i=0}^kS_{i}^kC_{x}^ii! \end{aligned} xk==i=0∑kSikxii=0∑kSikCxii!
那么我们变换一下,最后的公式就是:
∑ i = 1 n i k = ∑ i = 0 k S k i ( n + 1 ) i + 1 i + 1 \begin{aligned} \sum_{i=1}^ni^k=&\sum_{i=0}^kS_{k}^i\frac{(n+1)^{i+1}}{i+1} \end{aligned} i=1∑nik=i=0∑kSkii+1(n+1)i+1
大概就这样,简略的说一下。(溜了溜了XD