POJ 2417 Discrete Logging BSGS

http://poj.org/problem?id=2417

BSGS 大步小步法（ baby step giant step ）

sqrt( p )的复杂度求出 ( a^x ) % p = b % p中的x

https://www.cnblogs.com/mjtcn/p/6879074.html 

我的代码中预处理a==b和b==1的部分其实是不必要的，因为w=sqrt(p)(向上取整)，大步小步法所找的x包含从0到w^2。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<map>
using namespace std;
#define LL long long
LL p,a,b;
map<LL,LL>q;
int main(){
    while(~scanf("%lld%lld%lld",&p,&a,&b)){
        if(b==1){printf("0\n");continue;}
        if(a==b){printf("1\n");continue;}
        LL w=(LL)sqrt((double)p),x=1,y=1;if(w*w!=p)++w;
        q[b]=-1;
        for(int i=1;i<=w;++i){x=(x*a)%p;LL z=(b*x)%p;if(q[z]==0)q[z]=i;}
        bool f=0;
        for(int i=1;i<=w;++i){
            y=(y*x)%p;
            if(q[y]!=0){
                LL z=q[y];if(z==-1) z=0;
                z=(LL)i*w-z;f=1;
                printf("%lld\n",z);
                break;
            }
        }
        q[b]=0;x=1;
        for(int i=1;i<=w;++i){x=(x*a)%p;LL z=(b*x)%p;q[z]=0;}
        if(!f)printf("no solution\n");
    }
    return 0;
}