[BZOJ2480]=[BZOJ2995]Spoj3105 Mod

原题地址

P不为素数的离散对数问题.

作死全部变量用ll+用map,结果:

时限是10s，好险啊差点就T了…
其实只有两数相乘再取模那里会爆int，于是我们类型转换一下就可以了(具体看代码).
结果：

还是好慢OLZ不过至少快了2s+…

AC code:

#include <cstdio>
#include <cmath>
#include <map>
using namespace std;
typedef long long ll;

int gcd(int x,int y){
    return y?gcd(y,x%y):x;
}

int qpow(int x,int y,int p){
    if(!y) return 1;
    int z=qpow(x,y>>1,p);
    z=(ll)z*z%p;
    if(y&1) z=(ll)z*x%p;
    return z;
}

void exgcd(int a,int b,int& x,int& y){
    if(!b){
        x=1;y=0;
        return ;
    }
    exgcd(b,a%b,x,y);
    int xx=x,yy=y;
    x=yy;y=xx-(int)(a/b)*yy;
}

int getinv(int a,int p){
    int inv,tmp;
    exgcd(a,p,inv,tmp);
    if(inv<0) return -((-inv)%p-p);
    return inv%p;
}

int BSGS(int a,int p,int b){
    int q=(int)sqrt(p),inv=getinv(qpow(a,q,p),p);
    map<int,int> id;
    for(int i=0,tmp=1;i<q;i++,tmp=(ll)tmp*a%p){
        if(tmp==b) return i;
        id[tmp]=i;
    }
    for(int i=1;i<=q;i++){
        b=(ll)b*inv%p;
        if(id[b]) return id[b]+i*q;
    }
    return -1;
}

int solve(int a,int p,int b){
    for(int i=0,tmp=1;i<=100;i++,tmp=(ll)tmp*a%p) if(tmp==b) return i;
    int i=0,k=1;
    for(;;i++){
        int g=gcd(a,p);
        if(g==1){
            b=(ll)b*getinv(k,p)%p;
            break;
        }
        k=(ll)k*a/g%p;
        p/=g;
        if(b%g) return -1;
        b/=g;
    }
    return BSGS(a,p,b)+i;
}

int main(){
    while(1){
        int a,p,b,ans;
        scanf("%d%d%d",&a,&p,&b);
        if(!(a|p|b)) break;
        a%=p;b%=p;
        ans=solve(a,p,b);
        if(ans==-1) printf("No Solution\n");
        else printf("%d\n",ans);
    }

    return 0;
}