[TJOI2018]教科書般的褻瀆

題目連接:點這裏html

Solution:

咱們推導一下,不可貴出它讓咱們求的東西,咱們能夠發現殺死全部的隨從須要\(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

Code:

#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;
}
相關文章
相關標籤/搜索