【洛谷 P3807】【模板】卢卡斯定理/Lucas 定理

【洛谷 P3807 】【模板】卢卡斯定理/Lucas 定理

题目大意：

求出 n，m，p $C_{n+m}^m\mathrm{m}\mathrm{o}\mathrm{d}p$ 的值

思路：

我们发现 n,m,p 是一个数量级的。
一般如果 p 恒比 n,m 大的话我们可以 愉快的 这样搞：
$$C_n^m\mathrm{m}\mathrm{o}\mathrm{d}p=\frac{n!}{m!(n-m)!}\mathrm{m}\mathrm{o}\mathrm{d}p$$
那么对$(n-m)!和m!$求逆元就好了。
但是在这道题中$(n+m)\mathrm{m}\mathrm{o}\mathrm{d}p$有可能等于 0，那么费马小定理就不再适用，此时逆元为0
所以我们必须要将n,m控制在 0~p-1 的范围内，所以 卢卡斯定理/Lucas 定理 诞生了
$$C_n^m=C_{n/p}^{m/p}\ast C_{n\mathrm{m}\mathrm{o}\mathrm{d}p}^{m\mathrm{m}\mathrm{o}\mathrm{d}p}$$
证明在这里：

卢卡斯定理的证明

$C_{n/p}^{m/p}$这一段可以递归求解，另一段使用乘法逆元+阶乘求解
然后我们就可以愉快的递归+逆元求解组合数了
在这里证明如何线性求阶乘逆元
因为$$(n+1)\ast n!=(n+1)!$$
取倒数，有：$$\frac{1}{(n+1)\ast n!}=\frac{1}{(n+1)!}$$
左右两边同乘$(n+1)$ ，得：$$\frac{1}{n!}=\frac{1}{(n+1)!}\ast (n+1)$$
那么我们就可以在$O(1)$内倒推出$n!$的逆元，由此实现线性求阶乘逆元

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<vector>
#define r register
#define rep(i,x,y) for(r ll i=x;i<=y;++i)
#define per(i,x,y) for(r ll i=x;i>=y;--i)
using namespace std;
typedef long long ll;
const ll V=1e6+10;
ll t,n,m,p;
ll jc[V],inv[V];
ll qpow (ll x,ll y,ll p)
{
	ll ans=1;
	while(y)
	{
		if(y&1) ans=(ans*x)%p;
		x=(x*x)%p;
		y>>=1; 
	}
	return ans;
}
ll f(ll n,ll m)
{
	if(m>n) return 0;
	return (jc[n]*inv[m])%p*inv[n-m]%p;
}
ll C(ll n,ll m)
{
	if(m==0) return 1;
	return (C(n/p,m/p)*f(n%p,m%p)%p)%p; //递归求解组合数
}
int main()
{
	scanf("%lld",&t);
	while(t--)
	{
		scanf("%lld%lld%lld",&n,&m,&p);
		jc[0]=1;
		rep(i,1,p) jc[i]=(jc[i-1]*i)%p;
		inv[p-1]=qpow(jc[p-1],p-2,p);
		per(i,p-2,0) inv[i]=(inv[i+1]*(i+1))%p; //线性求解阶乘逆元
		printf("%lld\n",C(n+m,n));
	}
	return 0;
}