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;
}