Mod Tree HDU - 2815

Mod Tree

 HDU - 2815 

 题意：a^x≡n(mod p)，求x。

BSGS模板题~

需要注意的是n>p时显然无解

#include <bits/stdc++.h>
using namespace std;

#define ll long long
const int mod=99991;
ll head[mod],nex[mod];
ll cnt;
ll hs[mod],id[mod];
void init(){
    memset(head,-1,sizeof(head));
    cnt=0;
}
void insert_(ll x,ll i){
    hs[cnt]=x;id[cnt]=i;
    ll u=x%mod;
    nex[cnt]=head[u];
    head[u]=cnt++;
}
ll find_(ll x){
    ll u=x%mod;
    for(ll i=head[u];~i;i=nex[i]){
        if(hs[i]==x) return id[i];
    }
    return -1;
}
ll gcd(ll a,ll b){
    ll r=a%b;
    while(r){a=b;b=r;r=a%b;}
    return b;
}
//求解a^x≡b(mod c)，返回x，无解返回-1
ll BSGS(ll a,ll b,ll c){
    a%=c;b%=c;
    if(b==1) return 0;
    ll g=1,d=1,k=0;
    while((g=gcd(a,c))!=1){
        if(b%g) return -1;
        b/=g;c/=g;
        k++;
        d=d*(a/g)%c;
        if(b==d) return k;
    }
    ll m=ceil(sqrt(c*1.0)),q=1;
    for(ll i=0;i<=m;i++){
        if(i==0) insert_(b,i);
        else {
            q=q*a%c;
            insert_(q*b%c,i);
        }
    }
    for(ll i=1;i<=m;i++){
        d=d*q%c;
        ll j=find_(d);
        if(j!=-1) return i*m-j+k;
    }
    return -1;
}
int main(){
    ll k,p,n;
    while(scanf("%lld%lld%lld",&k,&p,&n)!=EOF){
        init();
        ll ans=-1;
        if(n<=p) ans=BSGS(k,n,p);
        if(ans==-1) puts("Orz,I can’t find D!");
        else printf("%lld\n",ans);
    }
}