Codeforces 622F The Sum of the k-th Powers ( 自然数幂和、拉格朗日插值法 )

题目链接

题意 : 就是让你求个自然数幂和、最高次可达 1e6 、求和上限是 1e9

分析 : 

题目给出了最高次 k = 1、2、3 时候的自然数幂和求和公式

可以发现求和公式的最高次都是 k+1

那么大胆猜测幂为 k 的自然数幂和肯定可以由一个最高次为 k+1 的多项式表示

不会证明，事实也的确如此

此时就变成了一个拉格朗日插值的模板题了

只要先算出 k+2 个值、然后就能确定最高次为 k+1 的多项式

套模板求值即可

当然自然数幂和不止这一种做法、例如伯努利数、斯特林数等

详细可参考 ==> Click here

#include<bits/stdc++.h>
#define LL long long
#define ULL unsigned long long

#define scs(i) scanf("%s", i)
#define sci(i) scanf("%d", &i)
#define scd(i) scanf("%lf", &i)
#define scIl(i) scanf("%I64d", &i)
#define scii(i, j) scanf("%d %d", &i, &j)
#define scdd(i, j) scanf("%lf %lf", &i, &j)
#define scIll(i, j) scanf("%I64d %I64d", &i, &j)
#define sciii(i, j, k) scanf("%d %d %d", &i, &j, &k)
#define scddd(i, j, k) scanf("%lf %lf %lf", &i, &j, &k)
#define scIlll(i, j, k) scanf("%I64d %I64d %I64d", &i, &j, &k)
#define sciiii(i, j, k, l) scanf("%d %d %d %d", &i, &j, &k, &l)
#define scdddd(i, j, k, l) scanf("%lf %lf %lf %lf", &i, &j, &k, &l)
#define scIllll(i, j, k, l) scanf("%I64d %I64d %I64d %I64d", &i, &j, &k, &l)

#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
#define lowbit(i) (i & (-i))
#define mem(i, j) memset(i, j, sizeof(i))

#define fir first
#define sec second
#define VI vector<int>
#define ins(i) insert(i)
#define pb(i) push_back(i)
#define pii pair<int, int>
#define VL vector<long long>
#define mk(i, j) make_pair(i, j)
#define all(i) i.begin(), i.end()
#define pll pair<long long, long long>

#define _TIME 0
#define _INPUT 0
#define _OUTPUT 0
clock_t START, END;
void __stTIME();
void __enTIME();
void __IOPUT();
using namespace std;
;
;

LL pow_mod(LL a, LL b)
{
    a %= mod;
    LL ret = ;
    while(b){
        ) ret = (ret * a) % mod;
        a = ( a * a ) % mod;
        b >>= ;
    }
    return ret;
}

namespace polysum
{
    #define rep(i,a,n) for (int i=a;i<n;i++)
    #define per(i,a,n) for (int i=n-1;i>=a;i--)
    ;
    LL a[D],f[D],g[D],p[D],p1[D],p2[D],b[D],h[D][],C[D];
    LL powmod(LL a,LL b){LL res=;a%=mod;assert(b>=);){)res=res*a%mod;a=a*a%mod;}return res;}
    LL calcn(int d,LL *a,LL n) { // a[0].. a[d]  a[n]
        if (n<=d) return a[n];
        p1[]=p2[]=;
        rep(i,,d+) {
            LL t=(n-i+mod)%mod;
            p1[i+]=p1[i]*t%mod;
        }
        rep(i,,d+) {
            LL t=(n-d+i+mod)%mod;
            p2[i+]=p2[i]*t%mod;
        }
        LL ans=;
        rep(i,,d+) {
            LL t=g[i]*g[d-i]%mod*p1[i]%mod*p2[d-i]%mod*a[i]%mod;
            ) ans=(ans-t+mod)%mod;
            else ans=(ans+t)%mod;
        }
        return ans;
    }
    void init(int M) {
        f[]=f[]=g[]=g[]=;
        rep(i,,M+) f[i]=f[i-]*i%mod;
        g[M+]=powmod(f[M+],mod-);
        per(i,,M+) g[i]=g[i+]*(i+)%mod;
    }
    LL polysum(LL m,LL *a,LL n) { // a[0].. a[m] \sum_{i=0}^{n-1} a[i]

        ;i<=m;i++) b[i]=a[i];
        b[m+]=calcn(m,b,m+);
        rep(i,,m+) b[i]=(b[i-]+b[i])%mod;
        ,b,n-);
    }
    LL qpolysum(LL R,LL n,LL *a,LL m) { // a[0].. a[m] \sum_{i=0}^{n-1} a[i]*R^i
        ) return polysum(n,a,m);
        a[m+]=calcn(m,a,m+);
        LL r=powmod(R,mod-),p3=,p4=,c,ans;
        h[][]=;h[][]=;
        rep(i,,m+) {
            h[i][]=(h[i-][]+a[i-])*r%mod;
            h[i][]=h[i-][]*r%mod;
        }
        rep(i,,m+) {
            LL t=g[i]*g[m+-i]%mod;
            ) p3=((p3-h[i][]*t)%mod+mod)%mod,p4=((p4-h[i][]*t)%mod+mod)%mod;
            ]*t)%mod,p4=(p4+h[i][]*t)%mod;
        }
        c=powmod(p4,mod-)*(mod-p3)%mod;
        rep(i,,m+) h[i][]=(h[i][]+h[i][]*c)%mod;
        rep(i,,m+) C[i]=h[i][];
        ans=(calcn(m,C,n)*powmod(R,n)-c)%mod;
        ) ans+=mod;
        return ans;
    }

}
LL arr[maxn];
int main(void){__stTIME();__IOPUT();

    int n, k;

    scii(n, k);

    ) return !printf("%d\n", n);

    polysum::init(k+);

    ; i<=k+; i++)
        arr[i] = pow_mod((LL)i, (LL)k);

    LL res = polysum::polysum(k+, arr, n+) - arr[];
    printf("%I64d\n", res);

__enTIME();;}

void __stTIME()
{
    #if _TIME
        START = clock();
    #endif
}

void __enTIME()
{
    #if _TIME
        END = clock();
        cerr<<"execute time = "<<(double)(END-START)/CLOCKS_PER_SEC<<endl;
    #endif
}

void __IOPUT()
{
    #if _INPUT
        freopen("in.txt", "r", stdin);
    #endif
    #if _OUTPUT
        freopen("out.txt", "w", stdout);
    #endif
}