三种方法放在洛谷上都过了(除了费马小定理求快速幂最后一个点TLE了),所以正确性是没问题的
才知道!!x1这个变量名!!是不能在LINUX下用的!!!
#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long
const int N=3e6+5;
//o(n)求逆元,要求模数必须为质数,并且所求的数小于模数
ll inv[N];
void getinv(int n,int p)
{
inv[1]=1;
for(int i=2;i<=n;i++)
inv[i]=(p-p/i)*inv[p%i]%p;
}
//o(logn) 用扩展欧几里德求逆元
void exgcd(int a,int b,ll &x,ll &y,ll &d)
{
if(!b) {
x=1,y=0,d=a;
return;
}
ll xx,yy;
exgcd(b,a%b,xx,yy,d);
x=yy,y=xx-(a/b)*yy;
}
//a在模m意义下的逆元 要求a,m互质 O(logn)
ll inverse(int a,int m)
{
ll x,y,d;
exgcd(a,m,x,y,d);//ax+py=1
if(d!=1) return -1;
return (x%m+m)%m;
}
//费马小定理求逆元 要求模数为质数 逆元=a^(p-2)
//这个常数明显要大,放到洛谷上要TLE最后一个点
ll Fermat(int a,int p)
{
ll base=a,ans=1;
for(int i=p-2;i;i>>=1,base=(base*base)%p)
if(i&1) ans=ans*base%p;
return ans;
}
//欧拉函数求逆元 类比费马小定理 x^(phi[m]-1)为x的逆元【所以还要求欧拉函数】
int main()
{
int n,p;
scanf("%d%d",&n,&p);
/* for(int i=1;i<=n;i++)
printf("%lld\n",Fermat(i,p));*/
/* for(int i=1;i<=n;i++)
printf("%lld\n",inverse(i,p));*/
/* getinv(n,p);
for(int i=1;i<=n;i++) printf("%lld\n",inv[i]);*/
return 0;
}
时间复杂度比较
O(n)求
O(nlogn) exgcd求
O(nlogn) 费马小定理求