【模板】卢卡斯定理/Lucas 定理
题目背景
这是一道模板题。
题目描述
给定整数 n , m , p n, m, p n,m,p 的值,求出 C n + m n m o d p C_{n + m}^n \bmod p Cn+mnmodp 的值。
输入数据保证 p p p 为质数。
注: C C C 表示组合数。
输入格式
本题有多组数据。
第一行一个整数 T T T,表示数据组数。
对于每组数据:
一行,三个整数 n , m , p n, m, p n,m,p。
输出格式
对于每组数据,输出一行,一个整数,表示所求的值。
样例 #1
样例输入 #1
2
1 2 5
2 1 5
样例输出 #1
3
3
提示
对于 100 % 100\% 100% 的数据, 1 ≤ n , m , p ≤ 1 0 5 1 \leq n, m, p \leq 10^5 1≤n,m,p≤105, 1 ≤ T ≤ 10 1 \leq T \leq 10 1≤T≤10。
模板
//lucas定理
#include<iostream>
#include<algorithm>
#include<cstring>
#define int long long
using namespace std;
const int N = 1e5+10;
int infac[N],fac[N];
int T;
int n,m,p;
int qmi(int a,int b,int p){
int res=1;
while(b){
if(b&1)res=res*a%p;
a=a*a%p;
b>>=1;
}
return res;
}
void init(){
fac[0]=infac[0]=1;
for(int i=1;i<N-10;i++){
fac[i]=fac[i-1]*i%p;
infac[i]=infac[i-1]*qmi(i,p-2,p)%p;
}
}
int C(int a,int b,int p){
int ans=0;
if(b>a)return 0;
return fac[a]%p*infac[b]%p*infac[a-b]%p;
}
int lucas(int a,int b,int p){
if(a<p&&b<p)return C(a,b,p);
return C(a%p,b%p,p)*lucas(a/p,b/p,p)%p;
}
signed main(){
cin>>T;
while(T--){
cin>>n>>m>>p;
init();
cout<<lucas(n+m,m,p)<<endl;
}
return 0;
}