模板—数学—Lucas

Posted 2021-02-15 yangsongyi

模板—数学—Lucas

Code:

#include <cstdio>
#include <algorithm>
using namespace std;
#define N 100010
int n,m,p,inv[N],powq[N];
int lucas(int n,int m)
{
	if(n<m) return 0;
	if(n<=p&&m<=p) return 1ll*powq[n]*inv[m]%p*inv[n-m]%p;
	return 1ll*lucas(n/p,m/p)*lucas(n%p,m%p)%p;
}
int main()
{
	int T;scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d",&n,&m,&p),n+=m; powq[0]=1,inv[p-1]=p-1;
		for(int i=1;i<=p;i++) powq[i]=1ll*powq[i-1]*i%p;
		for(int i=p-2;~i;i--) inv[i]=1ll*inv[i+1]*(i+1)%p;
		printf("%d
",lucas(n,m));
	}
}