题解 CF622F 【The Sum of the k-th Powers】

UPD:2020.7.28 修复了一些小锅qaq

题目链接

Solution CF622F The Sum of the k-th Powers

题目大意：给定(n,k),求(sum_{i=1}^ni^k)

拉格朗日插值

---

分析：首先类似于积分，自然数(k)次幂的前缀和(F(n))可以用一个(k + 1)次的多项式表达出来

那么我们就可以选取(k + 2)个点，用拉格朗日插值求出(F(n))

注意，如果随便选择点的话时间复杂度是(O(k^2))的，我们通过选取连续的点可以做到(O(k))

(F(n)=sum_{i=1}^{k+2}(i^k cdot frac{prod_{j eq i}n-j}{prod_{j eq i}i-j}))

首先观察分子，我们可以通过讨论(n)和(k+2)的大小关系，如果(n leq k + 2)那么我们就暴力算，否则插值，这样可以避免分子的符号讨论

对于分子，我们只需要预处理出(n-i quad i in [1,k+2])的前、后缀积就可以(O(1))求得

对于分母，观察发现实质上是两个阶乘之积，只不过要根据(k+2-i)的奇偶性来讨论符号

(i^k)可以快速幂在(O(logk))的时间内求得

总复杂度(O(klogk))

#include <cstdio>
#include <algorithm>
using namespace std;
const int mod = 1e9 + 7,maxk = 1e6 + 100;
typedef long long ll;
ll ans,fac[maxk],suf[maxk],pre[maxk],sum;
int n,k;
ll add(ll a,ll b){return (a + b) % mod;}
ll mul(ll a,ll b){return (a * b) % mod;}
ll qpow(ll a,ll b){
	ll res = 1,base = a;
	while(b){
		if(b & 1)res = mul(res,base);
		base = mul(base,base);
		b >>= 1;
	}
	return res;
}
ll inv(ll x){return qpow(x,mod - 2);}
ll pro(ll a,ll b){
	return mul(fac[b],inv(fac[a - 1]));
}
inline void init(){
	fac[0] = pre[0] = suf[k + 3] = 1;
	for(int i = 1;i <= k + 2;i++)fac[i] = mul(fac[i - 1],i);
	for(int i = 1;i <= k + 2;i++)pre[i] = mul(pre[i - 1],n - i);
	for(int i = k + 2;i >= 1;i--)suf[i] = mul(suf[i + 1],n - i);
}
int main(){
	scanf("%d %d",&n,&k);
	if(n <= k + 2){
		for(int i = 1;i <= n;i++)ans = add(ans,qpow(i,k));
		printf("%lld
",ans);
		return 0;
	}
	init();
	for(int i = 1;i <= k + 2;i++){
		ll s1 = 1,s2 = 1;
		if(i != 1)s1 = mul(s1,pre[i - 1]);
		if(i != k + 2)s1 = mul(s1,suf[i + 1]);

		if(i != 1)s2 = mul(s2,pro(1,i - 1));
		if(i != k + 2)s2 = mul(s2,((k + 2 - i) % 2) ? (mod - pro(1,k + 2 - i)) : pro(1,k + 2 - i));

		sum = add(sum,qpow(i,k));
		ans = add(ans,mul(sum,mul(s1,inv(s2))));
	}
	printf("%lld
",ans);
	return 0;
}