題目連接:點這裏html
咱們推導一下,不可貴出它讓咱們求的東西,咱們能夠發現殺死全部的隨從須要\(m+1\)張褻瀆c++
設\(S(x)=\sum_{i=1}^x i^{m+1}\),則題目要求的答案計算式爲:
\[ F(x)=\sum_{j=0}^m \sum_{i=j+1}^{m+1}S(a_i-a_j-1)-S(a_{i-1}-a_j) \]
特別的,咱們令\(a_{m+1}=n+1\)git
咱們發現第二個sigma能夠進行轉換:
\[ F(x)=\sum_{j=0}^m (S(n-a_{j})-\sum_{i=j+1}^m(a_i-a_j)^{m+1}) \]
無數神犇告訴咱們\(S(x)\)是一個以x爲自變量的m+2次多項式(dalao的證實),因此咱們能夠用拉格朗日插值來直接求ui
#include<bits/stdc++.h> #define int long long using namespace std; const int K=55; const int mod=1e9+7; int ans,num,pro,a[K],f[K],fac[K]={1}; int read(){ int x=0,f=1;char ch=getchar(); while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();} while(isdigit(ch)){x=x*10+ch-48;ch=getchar();} return x*f; } int quickpow(int x,int y){ int re=1; while(y){ if(y&1) re=re*x%mod; x=x*x%mod;y>>=1; }return re%mod; } int Lagrange(int n,int k){ pro=1,num=0; if(n<=k) return f[n]; for(int i=1;i<=k;i++) pro=pro*(n-i)%mod; for(int i=1;i<=k;i++){ int inv1=quickpow(n-i,mod-2); int inv2=quickpow((fac[i-1]%mod*fac[k-i])%mod,mod-2); int sign=(k-i)&1?-1:1; num=(num+sign*inv1*inv2%mod*f[i]%mod*pro%mod)%mod; }return (num+mod)%mod; } void solve(){ int n=read(),m=read(); for(int i=1;i<=m+3;i++) f[i]=(f[i-1]+quickpow(i,m+1))%mod; for(int i=1;i<=m;i++) a[i]=read(); sort(a+1,a+m+1);ans=0; for(int i=0;i<=m;i++){ ans=(ans+Lagrange(n-a[i],m+3))%mod; for(int j=i+1;j<=m;j++) ans=(ans+mod-quickpow(a[j]-a[i],m+1))%mod; }printf("%lld\n",(ans+mod)%mod); } signed main(){ for(int i=1;i<=K;i++) fac[i]=fac[i-1]*i%mod; int T=read();while(T--) solve(); return 0; }