擴展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;
}