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
",CRT());
	return 0;
}
原文地址:https://www.cnblogs.com/Bhllx/p/10659267.html