卢卡斯定理

卢卡斯定理

用途

卢卡斯(\(Lucas\))定理用于求当\(n\)、\(m\)较大时，\(C_n^m mod\;p(p为质数)\)的值。

描述

递归形式

对于

\[n,m \in \mathbb{Z},p为质数 \]

有

\[C_n^m mod\;p=C_{n\;mod\;p}^{m\;mod\;p}\cdot C_{n/p}^{m/p} mod\;p \]

非递归形式

对于

\[n,m \in \mathbb{Z},p为质数\\ 设m=\prod_{i=0}^{k}m_ip^i,n=\prod_{i=0}^{k}n_ip^i \]

有

\[C_n^m mod\;p=\prod_{i=0}^{k}C_{n_i}^{m_i}mod\;p \]

使用方法

对于比较大的组合数，我们先利用卢卡斯定理将其拆分，拆分所得到的每个较小的组合数则可以用更简单的方法求解。对于递归形式，拆分出的\(C_{n\;mod\;p}^{m\;mod\;p}\)还原成阶乘形式后，由于\(p\)为质数且除数与\(p\)互质，我们可以考虑运用费马小定理之类的方法求逆元进行计算。对于非递归形式，相当于把\(m\)、\(n\)均当做一个\(p\)进制下的数，其每一位也与\(p\)互质，也可以如此计算。

注意边界条件：当\(m=0\)时，\(C_n^m=1\)；当\(n<m\)时，\(C_n^m=0\)。阶乘可以进行预处理减少重复计算。

代码

//递归
int power(int b,int k)
{
	int base=1;
	while(k)
	{
		if(k&1)
			base=(long long)base*b%p;
		k>>=1;
		b=(long long)b*b%p;
	}
	return base;
}
int inv(int x)
{
	return power(x,p-2);
}
int C(int x,int y)
{
	return (long long)factorial[x]*inv((long long)factorial[y]*factorial[x-y]%p)%p;
}
int Lucas(int x,int y)
{
	if(!y)
		return 1;
	if(x%p<y%p)
		return 0;
	return C(x%p,y%p)*Lucas(x/p,y/p)%p;
}

//非递归
int Lucas(int x,int y)
{
	int result=1;
	while(x&&y)
	{
		if(x%p<y%p)
			return 0;
		result=(long long)result*C(x%p,y%p)%p;
		x/=p;
		y/=p;
	}
	return result;
}