HDU6372 sacul 【Lucas定理】

HDU6372

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL MOD=1e9+7;
const int N=15e5+10;
const int M=15e5+10;
LL fac[N],inv[N];
LL pod(LL x,LL n,LL MOD)
{
    LL ret=1;
    while(n)
    {
        if(n&1)ret=ret*x%MOD;
        x=x*x%MOD;
        n>>=1;
    }
    return ret;
}
void init(int n)
{
    fac[0]=fac[1]=inv[0]=inv[1]=1;
    for(LL i=2; i<=n; i++)fac[i]=fac[i-1]*i%MOD;

    inv[n]=pod(fac[n],MOD-2,MOD);

    for(LL i=n; i>1; i--)inv[i-1]=inv[i]*i%MOD;
}
int pri[N],tot=0;
bool check[N];
void dprr(int N)
{
    tot=0;
    for(int i=2; i<=N; i++)
    {
        if(!check[i])
        {
            pri[++tot]=i;
        }
        for(int j=1; j<=tot&&(LL)i*pri[j]<=N; j++)
        {
            check[i*pri[j]]=true;
            if(i%pri[j]==0)break;
        }
    }
}
LL C(int n,int m)
{
    return fac[n]*inv[m]%MOD*inv[n-m]%MOD;
}
LL solve(LL a,LL n)
{
    if(a==1)return n;
    LL ret=a*(pod(a,n,MOD)-1)%MOD*pod(a-1,MOD-2,MOD)%MOD;
    ret=(ret+MOD)%MOD;
    return ret;
}
int main()
{
    init(M-1);
    dprr(N-1);

    int TA,n,c,k,p;
    scanf("%d",&TA);
    while(TA--)
    {
        scanf("%d%d%d",&c,&n,&k);
        p=pri[c];
        LL ans=0,tem;
        for(int i=1; i<=k; i++)
        {
            tem=C(i+p,p-1);
            ans=(ans+solve(tem,1ll*n))%MOD;
        }
        ans=(ans+MOD)%MOD;
        printf("%lld\n",ans);
    }

}