bzoj 3122: [Sdoi2013]随机数生成器

 #include<cstdio>
 #include<iostream>
 #include<map>
 #include<cmath>
 #define ll long long
 using namespace std;
 ll T,p,a,b,c,x1,t;
 map<ll,ll> mp;
 ll exgcd(ll b,ll p,ll &x,ll &y)
 {
     if(!p)
       {
         x=;
         y=;
         return b;
       }
     ll t1=exgcd(p,b%p,x,y);
     ll t=x;
     x=y;
     y=t-b/p*y;
     return t1;
 }
 ll kuai(ll a,ll k,ll p)
 {
     ll ans=;
     for(;k;)
       {
         if(k%)
           ans=(ans*a)%p;
         k/=;
         a=(a*a)%p;
       }
     return ans;
 }
 ll pan(ll y,ll z,ll p)
 {
     mp.clear();
     ll m=ceil(sqrt(p)),t=;
     mp[]=m+;
     for(ll i=;i<m;i++)
         {
            t=t*y%p;
            mp[t]=i;
         }
     ll T=kuai(y,p-m-,p),ine=;
     for(ll k=;k<=m;k++)
        {
            ll i=mp[z*ine%p];
            if(i)
               {
                 if(i==m+)
                   i=;
                 return k*m+i+;
               }
           ine=ine*T%p;
        }
     return -;
 }
 ll solve()
 {
     if(x1==t)
       return ;
     if(a==)
       {
         if(b==t)
           return ;
         return -;
       }
     if(a==)
       {
         c=(t-x1+p)%p;
         ll x,y;
         ll t1=exgcd(b,p,x,y);
         if(c%t1)
           return -;
         c/=t1;
         x=x*c%p;
         for(;x<;x+=p);
         return x+;
       }
     if(a>)
       {
         c=kuai(a-,p-,p);
         ll x,y,d=(b*c+t)%p;
         ll t1=exgcd((x1+b*c)%p,p,x,y);
         if(d%t1)
           return -;
         d/=t1;
         x=x*d%p;
         for(;x<;x+=p);
         a%=p;
         x%=p;
         if(!a&&!x)
           return ;
         if(!a)
           return -;
         return pan(a,x,p);
       }
 }
 int main()
 {
     scanf("%lld",&T);
     for(int i=;i<=T;i++)
       {
         scanf("%lld%lld%lld%lld%lld",&p,&a,&b,&x1,&t);
         printf("%lld\n",solve());
       }
     return ;
 }

当 a==0时，判断b，当a==1时，exgcd否则BSGS