Discrete Logging POJ - 2417（BSGS）

Discrete Logging

 POJ - 2417 

题意：给P，B，N，求最小的L使得 B^L≡N (mod P)，其中P是素数。

Baby Step Giant Step

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
#define ll long long
using namespace std;
const int mod=99991;
ll head[mod],nex[mod];
ll cnt;
ll hs[mod],id[mod];
void init(){
    memset(head,-1,sizeof(head));
    cnt=0;
}
void insert_(ll x,ll i){
    hs[cnt]=x;id[cnt]=i;
    ll u=x%mod;
    nex[cnt]=head[u];
    head[u]=cnt++;
}
ll find_(ll x){
    int u=x%mod;
    for(int i=head[u];~i;i=nex[i]){
        if(hs[i]==x) return id[i];
    }
    return -1;
}
//求解A^x≡B(mod C),返回x，若无解返回-1
ll bsgs(ll a,ll b,ll c){
    if(b==1) return 0;
    ll m=ceil(sqrt(c*1.0));
    ll q=1;
    for(ll i=0;i<=m;i++){
        if(i==0) insert_(b%c,i);
        else{
            q=q*a%c;
            insert_(q*b%c,i);
        }
    }
    ll temp=1;
    for(ll i=1;i<=m;i++){
        temp=temp*q%c;
        ll j=find_(temp);
        if(j!=-1) return i*m-j;
    }
    return -1;
}
ll a,b,c;
int main(){
    while(scanf("%lld%lld%lld",&c,&a,&b)!=EOF){
        init();
        ll ans=bsgs(a,b,c);
        if(ans!=-1) printf("%lld\n",ans);
        else puts("no solution");
    }
}