Codeforces 622F (拉格朗日插值 费马小定理)

做这个题之前需要知道一些知识:

拉格朗日插值:n-1次多项式可以用n个点唯一确定,插值公式是:

费马小定理:a^(p-1)≡1(如果p是素数),也就是说模p时a和a^(p-2)互为逆元

关于模:当p是素数时

对于这道题可以发现对于每一个k次幂的积都可以用一个k+1次幂的多项式表示,于是我们可以构造出k+2个点来拟合这个多项式.拉格朗日插值公式在这里就是:

我们需要求的就是F(n),带入以后就是:

然后就是简单的预处理和取模了,仔细点就可以了

#include <bits/stdc++.h>
using namespace std;
#define maxn 1000005
#define mod  1000000007

long long fac[maxn], y[maxn];
long long n, k;

long long qpow (long long a, long long b) {
    a %= mod, b %= mod;
    long long ans = 1;
    while (b) {
        if (b&1)
            ans = ans*a%mod;
        a = a*a%mod;
        b >>= 1;
    }
    return ans;
}

int main () {
    //freopen ("in.txt", "r", stdin);
    cin >> n >> k;
    fac[0] = 1;
    for (long long i = 1; i <= k+2; i++) {
        fac[i] = fac[i-1]*i%mod; 
    } 
    if (k == 0) {
        cout << n << endl;
        return 0;
    }
    y[0] = 0;
    for (long long i = 1; i <= k+2; i++) {
        y[i] = y[i-1]+qpow (i, k); 
        y[i] %= mod;
    } 
    if (n <= k+2) {
        cout << y[n] << endl;
        return 0;
    }
    long long ans1 = 1;
    for (long long i = n-k-2; i <= n-1; i++) ans1 = ans1*i%mod;
    long long ans = 0;
    for (long long i = 1; i <= k+2; i++) {
        long long p1 = qpow (n-i, mod-2)%mod;
        long long p2 = qpow (fac[i-1]*fac[k+2-i]%mod, mod-2)%mod;
        long long sign = ((k+2-i)%2 == 1 ? -1 : 1);
        long long cur = y[i]*p1%mod*p2%mod*ans1%mod;
        cur *= sign;
        if (cur < 0)
            cur = (cur+mod)%mod;
        ans = (ans+cur)%mod;
    }
    cout << ans << endl;
    return 0;
}