51nod 1258 序列求和 V4(拉格朗日插值法)

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int N=50005;
long long a[N];
long long fac[N];
long long inv[N];
long long quickmod(long long a,long long b)
{
    long long ans=1;
    while(b)
    {
        if(b%2==1)
            ans=ans*a%mod;
        a=a*a%mod;
        b=b/2;
    }
    return ans;
}
int main()
{
    fac[0]=1;
    for(int i=1; i<N; i++)
        fac[i]=(fac[i-1])*i%mod;
    inv[N-1]=quickmod(fac[N-1],mod-2);
    for(int i=N-2; i>=0; i--)
        inv[i]=(inv[i+1]*(i+1))%mod;
    int t;
    scanf("%d",&t);
    while(t--)
    {
        long long n,k;
        scanf("%lld%lld",&n,&k);
        n%=mod;
        long long x=1;
        a[0]=0;
        for(int i=1; i<=k+2; i++)
        {
            a[i]=(a[i-1]+quickmod(i,k))%mod;
            x=x*(n-i)%mod;
        }
        x=(x+mod)%mod;
        if(n<=k+2)
        {
            printf("%lld\n",a[n]);
            continue;
        }
        long long ans=0;
        for(int i=1;i<=k+2;i++)
        {
            if((k+2-i)%2==1)
            ans=(ans-x%mod*quickmod((n-i+mod)%mod,mod-2)%mod*inv[k+2-i]%mod*inv[i-1]%mod%mod*a[i]%mod+mod)%mod;
            else
            ans=(ans+x%mod*quickmod((n-i+mod)%mod,mod-2)%mod*inv[k+2-i]%mod*inv[i-1]%mod%mod*a[i]%mod+mod)%mod;
            //ans=(ans+((((((x%mod*quickmod(n-i,mod-2)%mod)%mod)*inv[k+2-i]%mod)%mod*inv[i-1]%mod)%mod*quickmod(-1,k+2-i)%mod)%mod*a[i]%mod)%mod+mod)%mod;
        }
        printf("%lld\n",ans);
    }
    return 0;
}
/*
10
777033726439265809 40254
981401388
*/