为什么用数组存逆元会 WA 20pts
查看原帖
为什么用数组存逆元会 WA 20pts
779970
lunjiahao楼主2024/5/15 11:55

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;
}

求助

2024/5/15 11:55
加载中...