前言
前天又进行了一次校内胡策,内心很崩溃。。
自己还是数论太差了。。
知道的名词再多也没用,做题的时候不还是一样用不上。
Lucas定理
适用范围:取模数(以下用p表示)为质数时求解组合数。(若p不为质数则用扩展Lucas求解,即套用中国剩余定理,以后再补)
核心:C(n,m)=C(n/p,m/p)*C(n%p,m%p)
证明:不会,请自行度娘。
程序过程:
预处理1~p-1的阶乘%p,以及当前数在p进制下的逆元(若p为质数则可以用费马小定理求出,否则用扩展欧几里得求)。
将等式拆成C(n,m)=C(n/p,m/p)*C(n%p,m%p)的形式,将C(n%p,m%p)直接用预处理出的阶乘以及逆元求出,递归求解C(n/p,m/p)。
若出现n%p < m%p的情况,则说明答案%p后为0,直接退出。
CODE:
当前用的是费马小定理求逆元,注释掉的部分为用扩展欧几里得求逆元的方法。
inline ll pow(ll a,ll b,ll p)
{
ll ans=1;
a%=p;
for(;b;b>>=1,a=a*a%p)
if(b&1) ans=ans*a%p;
return ans;
}
inline void init()
{
A[1]=1;
for(int i=2;i<p;i++)
A[i]=A[i-1]*i%p,inv[i]=pow(i,p-2,p);
}
//void exgcd(ll a,ll b,ll &x,ll &y)
//{
// if(!b)
// {
// x=1,y=0;
// return;
// }
// exgcd(b,a%b,y,x);
// y-=(a/b)*x;
//}
//inline ll ni(ll a,ll b)
//{
// ll x,y;
// exgcd(a,b,x,y);
// x=(x%b+b)%b;
// return x;
return pow(a,b-2,p);
//}
inline ll lucas(ll n,ll m,ll p)
{
ll ans=1;
while(n&&m)
{
ll a=n%p,b=m%p;
if(a<b) return 0;
ans=(1ll*ans*A[a]*inv[A[b]]*inv[A[a-b]])%p;
n/=p,m/=p;
}
return ans;
}