bzoj2480 Spoj3105 Mod

题目

扩展BSGS，当时调了我很久，原因竟是p可以等于1，233。太惨了，这个可得好好注意一下呀。

大致思路和一般BSGS差不多，就是要多推几步，然后就一样了。

//ex_BSGS
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll a,b,p;
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(b==1)return 0;
/*--------------以下为多的部分-----------------*/
    ll cnt=0,t=1;
    for(ll g=gcd(a,p);g!=1;g=gcd(a,p))
    {
        if(b%g)return -1;
        p/=g,b/=g,t=t*a/g%p;
        cnt++;
        if(b==t)return cnt;
    }
/*------------------------------------------*/
    map <ll,ll> E;
    ll tmp=1,p1=int(sqrt(p*1.0)+1);
    for(ll i=0;i<p1;i++)E[(b*tmp)%p]=i,tmp=(tmp*a)%p;
    ll tmp1=t;
    for(ll i=1;i<=p1+1;i++)
    {
        tmp1=(tmp*tmp1)%p;
        if(E.count(tmp1))return (ll)i*p1-E[tmp1]+cnt;
    }
    return -1;
}
int main()
{
    freopen("in.txt","r",stdin);
    while(scanf("%lld%lld%lld",&a,&p,&b)!=EOF)
    {
        if(a==0&&p==0&&b==0)return 0;
        if(p==1)
        {
            if(a==0)printf("1\n");
            else printf("0\n");
            continue;
        }
        if(a==0&&b==0)
        {
            printf("1\n");
            continue;
        }
        ll ans=BSGS(a,b,p);
        if(ans==-1)printf("No Solution\n");
        else printf("%lld\n",ans);
    }
    return 0;
}