【Lucas定理】【扩展Lucas定理】模板

   

       Lucas定理主要用于求组合数取余，当模数mod为素数时，Lucas定理就能解决

       但当mod不是素数时，就需要用到扩展卢卡斯定理了

 

---

 

卢卡斯定理模板

ll PowMod(ll a,ll b,ll MOD){  //费马小定理求逆元
    ll ret=1;
    while(b){
        if(b&1) ret=(ret*a)%MOD;
        a=(a*a)%MOD;
        b>>=1;
    }
    return ret;
}
ll fac[100005];
ll Get_Fact(ll p){
    fac[0]=1;
    for(ll i=1;i<=p;i++)
        fac[i]=(fac[i-1]*i)%p;  //预处理阶乘
}
ll Lucas(ll n,ll m,ll p){
    ll ret=1;
    while(n&&m){
        ll a=n%p,b=m%p;
        if(a<b) return 0;
        ret=(ret*fac[a]*PowMod(fac[b]*fac[a-b]%p,p-2,p))%p;
        n/=p;
        m/=p;
    }
    return ret;
}
int main(){
        ll n,m,p;
        scanf("%lld%lld%lld",&n,&m,&p);
        Get_Fact(p);
        printf("%lld\n",Lucas(n,m,p));
    return 0;
}

---

 

扩展卢卡斯定理模板

ll n,m,mod;
ll pow(ll base,ll x,ll p){
    ll ans=1;
    while(x){
        if(x&1) ans=ans*base%p;
        base=base*base%p;
        x>>=1;
    }
    return ans;
}
ll exgcd(ll a,ll b,ll &x,ll &y){
    if(!b){
        x=1;y=0;
        return a;
    }
    ll r=exgcd(b,a%b,x,y);
    ll t=x;
    x=y;
    y=t-a/b*y;
    return r;
}
ll inv(ll a,ll b){
    ll x,y;
    return exgcd(a,b,x,y)==1?(x+b)%b:-1;
}
ll CRT(ll x,ll p){return x*inv(mod/p,p)%mod*mod/p%mod;}     //中国剩余定理
ll fac(ll x,ll p,ll pp){
    if(!x) return 1;
    ll res=1;
    for(int i=2;i<=p;i++) if(i%pp) res=res*i%p;
    res=pow(res,x/p,p);
    for(int i=2;i<=x%p;i++) if(i%pp) res=res*i%p;
    //递归处理n/pi
    return res*fac(x/pp,p,pp)%p;
}
ll C(ll n,ll m,ll p,ll pp){
    ll fn=fac(n,p,pp),fm=fac(m,p,pp),fnm=fac(n-m,p,pp);
    ll k=0;
    for(ll i=n;i;i/=pp) k+=i/pp;    //计算n!中pi的个数
    for(ll i=m;i;i/=pp) k-=i/pp;
    for(ll i=n-m;i;i/=pp) k-=i/pp;
    return fn*inv(fm,p)%p*inv(fnm,p)%p*pow(pp,k,p)%p;
}
ll exlucas(ll n,ll m){       //扩展Lucas
    ll t=mod,ans=0;
    for(int i=2;i*i<=mod;i++)
        if(t%i==0){
            ll p=1;
            while(t%i==0) p*=i,t/=i;
            ans=(ans+CRT(C(n,m,p,i),p))%mod;
        }
    if(t>1) ans=(ans+CRT(C(n,m,t,t),t)%mod);
    return ans%mod;
}
int main(){
    scanf("%lld%lld%lld",&n,&m,&mod);
    printf("%lld\n",exlucas(n,m));
    return 0;
}