BZOJ1467_Pku3243 clever Y_EXBSGS

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <map>
#include <math.h>
using namespace std;
typedef long long ll;
map<ll,int>mp;
void exgcd(ll a,ll b,ll &x,ll &y,ll &p) {
    if(!b) {x=1; y=0; p=a; return ;}
    exgcd(b,a%b,y,x,p);
    y-=a/b*x;
}
ll gcd(ll x,ll y) {
    return y?gcd(y,x%y):x;
}
ll EXBSGS(ll a,ll b,ll n) {
    if(n==1) return b==0?0:-1;
    if(a%n==0) return b==0?0:1;
    mp.clear();
    ll r,D=1,num=0,now,base=1;
    while((r=gcd(a,n))>1) {
        if(b%r) return -1;
        num++; b/=r; n/=r; D=D*a/r%n;
    }
    int i;
    for(i=0,now=1;i<num;i++,now=now*a%n)
        if(now==b) return i;
    ll m=ceil(sqrt(n));
    for(i=0;i<m;i++) {
        if(!mp.count(base)) mp[base]=i;
        base=base*a%n;
    }
    ll x,y,p;
    for(i=0;i<m;i++) {
        exgcd(D,n,x,y,p);
        x=(x*b%n+n)%n;
        if(mp.count(x)) return i*m+mp[x]+num;
        D=D*base%n;
    }
    return -1;
}
int main() {
    ll a,b,n;
    while(scanf("%lld%lld%lld",&a,&n,&b)!=EOF) {
        if(a==0&&b==0&&n==0) return 0;
        ll ans=EXBSGS(a,b,n);
        if(ans==-1) puts("No Solution");
        else printf("%lld\n",ans);
    }
}