【luoguP4720】【模板】扩展卢卡斯

快速阶乘与(扩展)卢卡斯定理

(p)为质数时

考虑 (n!~mod~p) 的性质

(n>>p)时,不妨将(n!)中的因子(p)提出来

(n!) 可以写成 (a*p^e)(a)(p)互质

如何求解(a)(e)

显然,(e=n/p+n/p^2+n/p^3+……)

因为(1)~(n)(n/p)(p)的倍数,贡献为(1)(n/p^2)(p^2)的倍数,贡献为(2)……

事实上,可以每次先将(1)$n$中$p$的倍数的因子提出来,$p,2p,3p...(n/p)p$就变成了$1,2,3...n/p$,$e+=n/p$,而$1$(n/p)这些数的(a)(e)又可以递归求解

再考虑(a)的求法:

(1)$n$中$p$的倍数拿走,递归处理求得$1$ (n/p)(a),再乘上剩下的数(1,2,3...p-1,p+1,p+2,...2p-1,2p+1,2p+2,...)就行了

而剩下的数在对(p)取模意义下是从(1)~(p-1)的循环,于是可以预处理(n< p)时的(n!)

剩下的数的积即为: (((p-1)!)^{n/p} imes(n~mod~p)!)

根据威尔逊定理,((p-1)!equiv -1 ~(mod~p)),可以省去快速幂

递归求解即可

卢卡斯定理

#include<iostream>
#include<cstring>
#include<cstdio>
#define int long long
using namespace std;

const int MAXN=100010;

int fact[MAXN];

inline int qpow(int x,int k,int p){
	int s=1%p;
	while(k){
		if(k&1) s=s*x%p;
		k>>=1;
		x=x*x%p;
	}
	return s;
}

inline int mod_fact(int n,int &e,int p){
	e=0;
	if(n<p) return fact[n];
	int res=mod_fact(n/p,e,p);
	e+=n/p;
	if(n/p%2!=0) return res*(p-fact[n%p])%p;	//((p-1)!)^(n/p)=(-1)^(n/p)
	return res*fact[n%p]%p;
}

inline int C(int n,int m,int p){
	if(m>n) return 0;
	int e1,e2,e3;
	int a1=mod_fact(n,e1,p),a2=mod_fact(m,e2,p),a3=mod_fact(n-m,e3,p);
	if(e1>e2+e3) return 0;	//p^(e1-e2-e3)是p的倍数,return 0
	return a1*qpow(a2*a3%p,p-2,p)%p;
}

int n,m,p;

signed main()
{
	int T;
	scanf("%lld",&T);
	while(T--){
		scanf("%lld%lld%lld",&n,&m,&p);
		fact[0]=1;
		for(int i=1;i<=p;++i)	//预处理阶乘
			fact[i]=fact[i-1]*i%p;
		printf("%lld
",C(n+m,m,p));
	}
	return 0;
}

(p)为任意数时

考虑将(p)分解为(p_1^{a_1}p_2^{a_2}p_3^{a_3}...p_k^{a_k})

分别求解(C(n,m)~mod ~ p_i^{a_i})

然后用(CRT)合并就可以了

求解(C(n,m)~mod~p_i^{a_i})时,仍然递归求解(1)~(n/p_i),但是(a)是对(p_i^{a_i})取模,而非对(p_i)取模

(p_i)的倍数的数乘积计算与上面也有不同,首先(p_i^{a_i})是合数,逆元只能用(exgcd)求了

因为是对(p_i^{a_i})取模,循环是(p_i^{a_i}-a_i),而且不能用威尔逊定理,需要先从(1)~(p_i^{a_i})暴力乘,再用快速幂直接计算,剩下的再暴力乘进去即可,此时预处理(fact[i])就不必要了

扩展卢卡斯

代码:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#define int long long
using namespace std;

const int MAXN=1000010;

inline int qpow(int x,int k,int p){
	int s=1;
	while(k){
		if(k&1) s=s*x%p;
		k>>=1;
		x=x*x%p;
	}
	return s;
}

inline int exgcd(int a,int b,int &x,int &y){
	if(!b){
		x=1; y=0;
		return a;
	}
	int t=exgcd(b,a%b,y,x);
	y-=a/b*x;
	return t;
}

inline int mod_fact(int n,int &e,int p,int MOD){
	if(!n) return 1;
	int res=1;
	for(int i=2;i<=MOD;++i)	//暴力累乘一个循环
		if(i%p) res=res*i%MOD;
	res=qpow(res,n/MOD,MOD);	//共有n/MOD个循环
	e+=n/p;
	for(int i=2;i<=n%MOD;++i)
		if(i%p) res=res*i%MOD;	//乘上循环外面的数
	return res*mod_fact(n/p,e,p,MOD)%MOD;	//递归处理n/p个p的倍数
}

inline int C(int n,int m,int p,int MOD){
	if(m>n) return 0;
	int e1=0,e2=0,e3=0;
	int a1=mod_fact(n,e1,p,MOD),a2=mod_fact(m,e2,p,MOD),a3=mod_fact(n-m,e3,p,MOD);
	if(e1-e2-e3>0){
		int temp=pow(p,e1-e2-e3);
		if(temp>=MOD) return 0;	//p^(e1-e2-e3)是p^ai的倍数
		a1=a1*temp%MOD;
	}
	int x,y; exgcd(a2*a3%MOD,MOD,x,y);
	return a1*x%MOD;
}

int n,m,p,a[110],b[110],cnt;

signed main()
{
	scanf("%lld%lld%lld",&n,&m,&p);
	int k=sqrt(p),x=p;
	for(int i=2;i<=k;++i)
	  if(x%i==0){
		int t=1;
		while(x%i==0)
			x/=i,t*=i;
		a[++cnt]=t;
		b[cnt]=C(n,m,i,t);
	}
	if(x!=1){
		a[++cnt]=x;
		b[cnt]=C(n,m,x,x);
	}
	int A=a[1],B=b[1];
	for(int i=2;i<=cnt;++i){
		int k1,k2;
		int g=exgcd(A,a[i],k1,k2);
		int t=A;
		A=A*a[i]/g;
		B=(t*k1%A*(b[i]-B)/g%A+B)%A;
	}
	printf("%lld
",(B+A)%A);
	return 0;
}
原文地址:https://www.cnblogs.com/yjkhhh/p/11719928.html