1. 程式人生 > >洛谷 P3807 【模板】盧卡斯定理

洛谷 P3807 【模板】盧卡斯定理

cnblogs col define 包含 main radius adg log lag

題目背景

這是一道模板題。

題目描述

給定n,m,p(1\le n,m,p\le 10^51n,m,p105)

C_{n+m}^{m}\ mod\ pCn+mm? mod p

保證P為prime

C表示組合數。

一個測試點內包含多組數據。

輸入輸出格式

輸入格式:

第一行一個整數T(T\le 10T10),表示數據組數

第二行開始共T行,每行三個數n m p,意義如上

輸出格式:

共T行,每行一個整數表示答案。

輸入輸出樣例

輸入樣例#1: 復制
2
1 2 5
2 1 5
輸出樣例#1: 復制
3
3

盧卡斯定理
$C(n,m)%p=C(n%p,m%p)*C(n/p,m/p)$
對於這道題來說,p是素數,解逆元的時候用快速冪

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<cmath>
 4 #include<algorithm>
 5 #define LL long long 
 6 using namespace std;
 7 const LL MAXN=1e6+10;
 8 const LL INF=0x7fffff;
 9 inline LL read()
10 {
11 char c=getchar();LL flag=1,x=0; 12 while(c<0||c>9) {if(c==-) flag=-1;c=getchar();} 13 while(c>=0&&c<=9) x=x*10+c-48,c=getchar();return x*flag; 14 } 15 LL js[MAXN]; 16 LL fastpow(LL a,LL p,LL mod) 17 { 18 LL base=1; 19 while(p) 20 {
21 if(p&1) base=(base*a)%mod; 22 a=(a*a)%mod; 23 p>>=1; 24 } 25 return base; 26 } 27 LL C(LL n,LL m,LL mod) 28 { 29 if(m>n) return 0; 30 return js[n]*fastpow(js[m],mod-2,mod)*fastpow(js[n-m],mod-2,mod)%mod; 31 } 32 LL Lucas(LL n,LL m,LL mod) 33 { 34 if(m==0) return 1; 35 else return C(n%mod,m%mod,mod)*(Lucas(n/mod,m/mod,mod))%mod; 36 } 37 int main() 38 { 39 LL T=read(); 40 js[0]=1; 41 while(T--) 42 { 43 LL n=read(),m=read(),mod=read(); 44 for(LL i=1;i<=mod;i++) js[i]=(js[i-1]*i)%mod; 45 printf("%lld\n",Lucas(n+m,m,mod)%mod); 46 } 47 return 0; 48 }

洛谷 P3807 【模板】盧卡斯定理