Lucas&&Exlucas
Lucas&&Exlucas
Lucas和Exlucas可以求模p意义下大数的组合数。
先考虑p为质数的情况,那么直接上Lucas定理即可。
Lucas 定理基本内容:
\[C_n^m=C_{n\ mod\ p}^{m\ mod\ p}*C_{n/p}^{m/p}\ (mod\ p)\ p是质数
\]
对于Lucas的实现直接递归处理即可,注意特判n<m的情况。
如果p不为质数,那么用Exlucas求解。
实际上,Exlucas和Lucas定理没有关系……
由于p不为质数,所以可以考虑把p质因数分解:
\[p=p_1^{c_1}*p_2^{c_2}...p_t^{c_t}
\]
那么可以问题可以转化成求解下列每一个式子的值,然后用CRT合并答案即可:
\[C_n^m\ mod\ p_1^{c_1}\\
C_n^m\ mod\ p_2^{c_2}\\
...\\
C_n^m\ mod\ p_t^{c_t}\\
\]
继续转化,把组合数写成阶乘形式,相当于要求:
\[\frac{n!}{m!·(n-m)!}\ mod\ p^k
\]
发现直接做不好做,逆元问题都不好解决,考虑把左边每一项中的因子p给提出来:
\[\frac{\frac{n!}{p^{a_1}}}{\frac{m!}{p^{a_2}}*\frac{(n-m)!}{p^{a3}}}*p^{a^1-a^2-a^3}\ mod\ p^k
\]
这样逆元就可以直接用Exgcd求了。
那么现在重点解决的问题又变成了:
\[n!\ mod\ p^k\ \ \ \ \ \ 其中n还要除去所有的因子p
\]
举个例子:n=22,p=3,k=2
把其中所有p(也就是3)的倍数提取出来,得到:
\[22!=3^7×(1×2×3×4×5×6×7)×(1×2×4×5×7×8)×(10×11×13×14×16×17)×(19×20×22) \]
可以发现:
\[(1×2×4×5×7×8)\equiv(10×11×13×14×16×17)\ (mod\ 3)
\]
所以对于这种一段一段的可以直接处理,最后求一下它(n/p^k)次方即可。
对于3的次方不需要考虑,因为在外面会乘上来。
那么为什么不把3,6彻底分解呢?
这是为了递归的方便,可以发现存在一项7!,也就是(n/p)!,可以递归处理。
所以在一层层递归中,每次每个含有因数p的数提且仅提出一个p,那么可以实现递归,且最后可以把所有p都提出!
然后就可以开心的求出组合数了O(∩_∩)O!
IL int qpow(int x,int p,int mod) {
RG int ans=1;
for(;p;p>>=1,x=x*x%mod)
if(p&1) ans=ans*x%mod;
return ans;
}
IL int exgcd(int &x,int &y,int a,int b) {
if(!b) return x=1,y=0,a;
RG int gcd=exgcd(x,y,b,a%b),z=x;
x=y,y=z-y*(a/b);
return gcd;
}
IL int Fac(int a,int p,int pk) {
if(a==0) return 1;
RG int i,ans=1;
for(i=1;i<pk;++i)
if(i%p) ans=ans*i%pk;
ans=qpow(ans,a/pk,pk);
for(i=1;i<=a%pk;++i)
if(i%p) ans=ans*i%pk;
return ans*Fac(a/p,p,pk)%pk;
}
IL int C(int a,int b,int p,int pk) {
if(a<b) return 0;
RG int i,x,y,k=0,f1=Fac(a,p,pk),f2=Fac(b,p,pk),f3=Fac(a-b,p,pk);
exgcd(x,y,f2,pk),x=(x+pk)%pk,f2=x;
exgcd(x,y,f3,pk),x=(x+pk)%pk,f3=x;
for(i=a/p;i;i/=p) k+=i;
for(i=b/p;i;i/=p) k-=i;
for(i=(a-b)/p;i;i/=p) k-=i;
return f1*f2%pk*f3%pk*qpow(p,k,pk)%pk;
}
IL int CRT() {
RG int i,x,y,ans=0;
for(i=1;i<=tot;++i) {
exgcd(x,y,p/Pk[i],Pk[i]),x=(x+Pk[i])%Pk[i];
ans=(ans+a[i]%p*(p/Pk[i])%p*x%p)%p;
}
return ans;
}
signed main()
{
RG int i,j,x;
n=gi(),m=gi(),p=gi();
for(i=2,x=p;i*i<=x;++i) {
if(x%i==0) {
pr[++tot]=i;
while(x%i==0) ++cnt[tot],x/=i;
}
}
if(x>1) pr[++tot]=x,cnt[tot]=1;
for(i=1;i<=tot;++i) {
for(j=1,x=1;j<=cnt[i];++j) x=x*pr[i];
a[i]=C(n,m,pr[i],Pk[i]=x);
}
printf("%lld\n",CRT());
return 0;
}