rt,这个是用数组存逆元的代码:
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+5;
int T,fac[N],inv[N];
int pow(int a,int b,int p)
{
int res=1;
a%=p;
while(b)
{
if(b&1) res=res*a%p;
a=a*a%p;
b>>=1;
}
return res;
}
int C(int n,int r,int p)
{
if(r>n) return 0ll;
return fac[n]*inv[n-r]%p*inv[r]%p;
}
int Lucas(int n,int r,int p)
{
if(!r) return 1ll;
return C(n%p,r%p,p)*Lucas(n/p,r/p,p)%p;
}
signed main()
{
scanf("%lld",&T);
fac[0]=1ll;
while(T--)
{
int n,m,p;
scanf("%lld%lld%lld",&n,&m,&p);
for(int i=1;i<=p;i++)
{
fac[i]=fac[i-1]*i%p;
inv[i]=pow(fac[i],p-2,p);
}
printf("%lld\n",Lucas(n+m,n,p));
}
return 0;
}
这是不用数组存逆元直接求的 AC 代码:
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+5;
int T,fac[N];
int pow(int a,int b,int p)
{
int res=1;
a%=p;
while(b)
{
if(b&1) res=res*a%p;
a=a*a%p;
b>>=1;
}
return res;
}
int C(int n,int r,int p)
{
if(r>n) return 0ll;
return fac[n]*pow(fac[n-r],p-2,p)%p*pow(fac[r],p-2,p)%p;
}
int Lucas(int n,int r,int p)
{
if(!r) return 1ll;
return C(n%p,r%p,p)*Lucas(n/p,r/p,p)%p;
}
signed main()
{
scanf("%lld",&T);
fac[0]=1ll;
while(T--)
{
int n,m,p;
scanf("%lld%lld%lld",&n,&m,&p);
for(int i=1;i<=p;i++)
fac[i]=fac[i-1]*i%p;
printf("%lld\n",Lucas(n+m,n,p));
}
return 0;
}
求助