hdu 2815 Mod Tree (exBSGS)

http://acm.hdu.edu.cn/showproblem.php?pid=2815

//解 K^D ≡ N mod P
#include<map>
#include<cmath>
#include<cstdio>
#include<iostream>

using namespace std; 

map<int,int>mp;

typedef long long LL;

void read(int &x)
{
    x=; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*+c-''; c=getchar(); }
}    

int Pow(int a,int b,int mod)
{
    int res=;
    for(;b;a=1LL*a*a%mod,b>>=)
        if(b&) res=1LL*res*a%mod;
    return res;
}

int gcd(int a,int b)
{
    return !b ? a : gcd(b,a%b);
}

LL exBSGS(int A,int B,int C)
{
    if(B==) return ;
    int k=,tmp=,d;
    while()
    {
        d=gcd(A,C);
        if(d==) break;
        if(B%d) return -;
        B/=d; C/=d;
        tmp=1LL*tmp*(A/d)%C;
        k++;
        if(tmp==B) return k;
    }
    mp.clear();
    int m=ceil(sqrt(1.0*C));
    mp[B]=;
    int mul=B;
    for(int j=;j<=m;++j)
    {
        mul=1LL*mul*A%C;
        mp[mul]=j;
    }
    int am=Pow(A,m,C);
    mul=tmp;
    for(int j=;j<=m;++j)
    {
        mul=1LL*mul*am%C;
        if(mp.find(mul)!=mp.end()) return 1LL*j*m-mp[mul]+k;
    }
    return -;
}

int main()
{
    int k,p,n;
    LL d;
    while(scanf("%d",&k)!=EOF)
    {
        read(p); read(n);
        if(n>=p)
        {
            puts("Orz,I can’t find D!");
            continue;
        }
        d=exBSGS(k,n,p);
        if(d==-) puts("Orz,I can’t find D!");
        else cout<<d<<'\n';
    }
}