#2578. 「TJOI2018」教科书般的亵渎 拉格朗日插值

题解：

显然这题就是求一个自然数幂和减去一些东西，自然数幂和是可以用拉格朗日插值求的，这个东西其实很简单，自己推也能推出来，主要是第一次写，记录一下。代码其实可以优化成$O(Tm^3)$，但是过了就懒得优化了。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define LL long long
const int Maxm=58;
const LL mod=1000000007LL;
LL Pow(LL x,LL y)
{
    if(!y)return 1;
    if(y==1)return x%mod;
    LL t=Pow(x,y>>1),re=t*t%mod;
    if(y&1)re=re*(x%mod)%mod;
    return re;
}
LL n,a[Maxm];int m,k;
LL X[Maxm],Y[Maxm];
LL calc(LL x)
{
    LL re=0;
    for(int i=1;i<=k+2;i++)
    {
        LL tmp=1;
        for(int j=1;j<=k+2;j++)
        {
            if(i==j)continue;
            tmp=tmp*((x-X[j]+mod)%mod)%mod*Pow((X[i]-X[j]+mod)%mod,mod-2)%mod;
        }
        tmp=tmp*Y[i]%mod;
        re=(re+tmp)%mod;
    }
    return re;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld%d",&n,&m);
        for(int i=1;i<=m;i++)scanf("%lld",&a[i]);
        sort(a+1,a+1+m);
        for(int i=m;i;i--)
        {
            if(a[i]==n)n--,m--;
            else break;
        }
        k=m+1;
        LL sum=0;
        for(int i=1;i<=k+2;i++)
        {
            sum=(sum+Pow(i,k))%mod;
            X[i]=i,Y[i]=sum;
        }
        LL ans=0,tmp=0,tn=n;a[0]=0;
        for(int i=0;i<k;i++)
        {
            ans=(ans+calc(n))%mod;
            for(int j=1;j<=m;j++)
            if(a[j]>tmp)ans=(ans-Pow(a[j]-tmp,k)+mod)%mod;
            tmp=a[i+1];
            n=(tn-tmp+mod)%mod;
        }
        printf("%lld\n",ans);
    }
}