P3807 【模板】卢卡斯定理

本题采用卢卡斯定理

公式如图。证明繁杂，直接练习写模板。

本题重点：1.模板 2.求解的式子中是c(n+m,m)而不是c(n,m)。这与组合数定义的c(n,m)不同，计算时要注意 3.本题多数变量需要使用LL定义或者强制转换，出错时需要查询是否都正确LL

lucas函数利用卢卡斯定理分解问题

LL lucas(LL n,LL m){
	if(n<p&&m<p)return C(n,m)%p;
	else return (LL)C(n%p,m%p)*lucas(n/p,m/p)%p;
}

而求组合数的函数C利用

将求组合数转化为m次迭代,分母部分是用逆元来求解

int p;
int qmul(int a,int b){
	int ans=1;
	while(b){
		if(b&1)(LL)ans=ans*a%p;
		a=(LL)a*a%p;
		b>>=1;
	}
	return ans;
}
int C(int n,int m){
	int ans=1;
	for(int i=1;i<=m;i++)
	{
		ans=(LL)ans*(n-i+1)%p;
		ans=(LL)ans*qmul(i,p-2)%p;
	}
	return ans;
}

综上，代码如下

#include<bits/stdc++.h>
using namespace std;
const int N=1e5;
typedef long long LL;
int p;
int qmul(int a,int b){
	int ans=1;
	while(b){
		if(b&1)(LL)ans=ans*a%p;
		a=(LL)a*a%p;
		b>>=1;
	}
	return ans;
}
int C(int n,int m){
	int ans=1;
	for(int i=1;i<=m;i++)
	{
		ans=(LL)ans*(n-i+1)%p;
		ans=(LL)ans*qmul(i,p-2)%p;
	}
	return ans;
}
LL lucas(LL n,LL m){
	if(n<p&&m<p)return C(n,m)%p;
	else return (LL)C(n%p,m%p)*lucas(n/p,m/p)%p;
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		LL n,m;
		scanf("%lld%lld%d",&n,&m,&p);
		cout<<lucas(n+m,m)<<endl;
	}
	return 0;
}