卢卡斯定理

最新推荐文章于 2022-05-21 15:35:21 发布

Hypoc_

最新推荐文章于 2022-05-21 15:35:21 发布

阅读量510

点赞数

分类专栏：数论

本文链接：https://blog.csdn.net/a_forever_dream/article/details/102656637

版权

数论专栏收录该内容

25 篇文章 10 订阅

订阅专栏

介绍

这是个用来求解大组合数取模意义下的值的算法。

要求取模的数是个质数。

算法

卢卡斯定理就是这样一条柿子：
${n\choose m}\equiv{n \bmod p \choose m \bmod p}{n\div p \choose m\div p} \pmod p$

（注意， $\div$ 是整除。）

当我们有了这条结论之后，我们可以对 ${n\div p \choose m\div p}$ 这部分不断使用卢卡斯定理，对 $\bmod p \choose m \bmod p}$ 这一部分计算组合数。如果提前 $O (n)$ 预处理阶乘以及阶乘的逆元，就可以做到 $O(log_pn)$ 计算大组合数。

证明

显然 $n$ 可以表示成 $a p + b$ ， $m$ 可以表示为 $c p + d$ ，那么上面的柿子可以变为：
$\choose cp+d}\equiv{b \choose d} {a \choose c} \pmod p$

那么有：
$\begin{aligned} (1+x)^n&\equiv(1+x)^{ap}\times (1+x)^b \pmod p\\ (1+x)^n&\equiv((1+x)^p)^a\times (1+x)^b \pmod p\\ \end{aligned}$

我们考虑转化里面 $1+x)^p$ 这一项，用神奇的二项式定理

$(1+x)^p \equiv \sum_{i=0}^p {p\choose i} x^i \pmod p\\$

当 $i = 0$ 时，这一项的值为 $1$ 。

当 $i = p$ 时，这一项的值为 $x^p$ 。

当 $i\in(0,p)$ 时， ${p\choose i}=\frac {p!} {i!(p-i)!}$ ，由于 $p$ 是个质数，所以这个系数肯定是 $p$ 的倍数，于是这一项为 $0$ 。

这里就是普通卢卡斯定理的限制—— $p$ 必须是质数——的缘由，如果 $p$ 不是质数，那么就不能保证这个系数是 $p$ 的倍数。而如果 $p$ 是个质数，那么当 $i\in (0,p)$ 时，分母部分不可能含有 $p$ 这个因子，而分子里肯定含有 $p$ 这个因子，所以总的这个系数肯定是 $p$ 的倍数。

所以有
$\sum_{i=0}^p {p\choose i}x^i\equiv 1+x^p \pmod p$

带回去有：
$\begin{aligned} (1+x)^n&\equiv (1+x^p)^a\times (1+x)^b \pmod p\\ \sum_{i=0}^n {n\choose i}x^i&\equiv \sum_{i=0}^a {a\choose i} x^{ip} \times \sum_{i=0}^b {b\choose i} x^i \pmod p \end{aligned}$

左边这个多项式中， $x^{cp+d}$ 的系数为 ${n\choose cp+d}$ ，即 ${ap+b\choose cp+d}$ 。

因为 $b < p$ ，所以右边只有 $x^{cp}$ 和 $x^d$ 这两项的积为 $x^{cp+d}$ ，而这两项的系数之积为 ${a\choose c}{b\choose d}$ ，所以有：
$\choose cp+d}\equiv{b \choose d} {a \choose c} \pmod p$

得证。

模板题传送门

AC代码：

#include <cstdio>
#include <cstring>
#define ll long long

int t,n,m,p;
ll fac[100010];
ll ksm(ll x,int y)
{
	ll re=1,tot=x;
	while(y)
	{
		if(y&1)re=re*tot%p;
		tot=tot*tot%p;
		y>>=1;
	}
	return re;
}
ll C(int x,int y)
{
	if(x<y)return 0;
	return fac[x]*ksm(fac[y]*fac[x-y]%p,p-2)%p;
}
ll lucas(ll x,ll y)
{
	if(y==0)return 1ll;
	return lucas(x/p,y/p)*C(x%p,y%p)%p;
}

int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d %d %d",&n,&m,&p);
		fac[0]=1;
		for(int i=1;i<=p;i++)
		fac[i]=fac[i-1]*i%p;
		printf("%lld\n",lucas(n+m,m));
	}
}