hdu 2815 : Mod Tree 【扩展BSGS】

题目链接

直接用模板好了。实在不行，反正有队友啊~~~~

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

map<LL,LL>mp;
LL qpow(LL x,LL n,LL mod)        //求x^n%mod
{
    LL ret=;
    )
    {
        ) ret=ret*x%mod;
        x=x*x%mod;
    }
    return ret;
}
LL gcd(LL a, LL b)
{
    return b? gcd(b,a%b):a;
}

LL BSGS(LL a,LL b,LL p)
{
    a%=p;
    b%=p;
    // if (a==0&&b==0) return 0;
    // if (a==0) return -1;
    ) ;
    ;
    LL t=;
    ; g=gcd(a,p))
    {
        ;
        p/=g;
        b/=g;
        t=t*a/g%p;
        cnt++;
        if(b==t) return cnt;
    }
    mp.clear();
    int m=ceil(sqrt(double(p)));
    LL base=b ;
    ; i<m; i++)
    {
        mp[base]=i;
        base=base*a%p;
    }
    base=qpow(a,m,p);
    LL ret=t;
    ; i<=m+; i++)
    {
        ret=ret*base%p;
        if(mp.count(ret)) return i*m-mp[ret]+cnt;
    }
    ;
}

int main()
{
    LL K,D,N,P;
    while(cin>>K>>P>>N)
    {
        LL ans=BSGS(K,N,P);
         || P== || N>=P)
            puts("Orz,I can’t find D!");
        else
            cout<<ans<<endl;
    }
}