题目传送门:https://www.luogu.org/problemnew/show/P3811
题意:
给出n,p,求1~n在(mod p)意义下的逆元。
思路1(66分,无任何优化):
发现p为质数,费马小定理直接上。
代码1:
#include<cstdio>
#define LL long long
LL n,p;
LL dg(LL x,LL k)
{
if(!k) return 1;
LL o=dg(x,k>>1)%p;
return (k&1)?o*o%p*x%p:o*o%p;
}
LL inv(LL x,LL p)
{
return dg(x,p-2);
}
int main()
{
scanf("%lld %lld",&n,&p);
for(int i=1;i<=n;i++)
printf("%lld\n",inv(i,p));
}
思路2(83分,无任何优化):
考虑扩展欧几里得。
代码2:
#include<cstdio>
#define LL long long
LL n,p;
LL x,y;
LL exgcd(LL a,LL b,LL &x,LL &y)
{
if(!b)
{
x=1;
y=0;
return a;
}
LL tmp=exgcd(b,a%b,y,x);
y-=a/b*x;
return tmp;
}
LL inv(LL x,LL y)
{
LL A=x,B=y,C=1,k=exgcd(A,B,x,y),p=B/k;
return (x*(C/k)%p+p)%p;
}
int main()
{
scanf("%lld %lld",&n,&p);
for(LL i=1;i<=n;i++)
printf("%lld\n",inv(i,p));
}
思路3(100分):
考虑线性求逆元。
代码3:
#include<cstdio>
int n,p;
int inv[3000001];
int main()
{
scanf("%d %d",&n,&p);
inv[1]=1;
printf("1\n");
for(int i=2;i<=n;i++)
{
inv[i]=(long long)(p-p/i)*inv[p%i]%p;
printf("%d\n",inv[i]);
}
}
总结:
PS:证明都在我的文章里(就不给你们看了)。
求一个数的逆元,选择扩展欧几里得(不要求p为质数),最快;
多个数的逆元,选择线性的方法(不要求p为质数),最快。
费马小定理再求逆元方面还是不建议使用(但代码较扩展欧几里得简洁,但是略慢,大概是一个常数的时间)。