题目背景
这是一道模板题。
题目描述
给定(n,m,p( 1le n,m,ple 10^5))
求 (C_{n+m}^{m} mod p)
保证 (p) 为prime
(C) 表示组合数。
一个测试点内包含多组数据。
输入输出格式
输入格式:
第一行一个整数 (T( Tle 10 )),表示数据组数
第二行开始共 (T) 行,每行三个数 (n,m,p),意义如上
输出格式:
共T行,每行一个整数表示答案。
输入输出样例
输入样例#1:
2
1 2 5
2 1 5
输出样例#1:
3
3
题解
卢卡斯定理的模板题
[igg(egin{matrix} m \ n end{matrix}igg) mod p=igg(egin{matrix} m/p \ n/p end{matrix}igg)igg(egin{matrix} mmod p \ n~mod p end{matrix}igg) mod p
]
具体应用就是预处理模数的fac和inv,然后两个组合数中后面那个肯定可以用预处理的算,前面那个如果 (n) 或 (m) 还是比模数大,就一直递归下去。注意,当 (n) 小于 (m) 时,返回0
证明看这里
那这道题就裸啦
#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
const int MAXP=100000+10;
ll fac[MAXP],inv[MAXP];
template<typename T> inline void read(T &x)
{
T data=0,w=1;
char ch=0;
while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
if(ch=='-')w=-1,ch=getchar();
while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
x=data*w;
}
template<typename T> inline void write(T x,char ch=' ')
{
if(x<0)putchar('-'),x=-x;
if(x>9)write(x/10);
putchar(x%10+'0');
if(ch!=' ')putchar(ch);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline ll qexp(ll a,ll b,int n)
{
ll res=1;
while(b)
{
if(b&1)res=res*a%n;
a=a*a%n;
b>>=1;
}
return res;
}
inline void init(int p)
{
fac[0]=1;
for(register int i=1;i<p;++i)fac[i]=1ll*fac[i-1]*i%p;
inv[p-1]=qexp(fac[p-1],p-2,p);
for(register int i=p-2;i>=0;--i)inv[i]=1ll*inv[i+1]*(i+1)%p;
}
inline ll C(ll n,ll m,int p)
{
if(n<m)return 0;
if(n<p&&m<p)return fac[n]*inv[m]%p*inv[n-m]%p;
else return C(n/p,m/p,p)*C(n%p,m%p,p)%p;
}
int main()
{
int T;
read(T);
while(T--)
{
int n,m,p;
read(n);read(m);read(p);
init(p);
write(C(n+m,m,p),'
');
}
return 0;
}