浅谈大步小步算法

解决什么？

先看一个式子

x的y次方≡z (mod p)

（z是质数）
现在只知道x和z，要求y。
大步小步算法（BSGS，Baby Steps Giant Steps）就是解决这个问题。

代码

Hash、map都可以。

idy002 orz

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cassert>
#include<cmath>
#include<map>
using namespace std;
const int S=100007;
struct Hash 
{
    int head[S],dest[S][2],last[S],etot;
    void init() 
    {
        memset(head,0,sizeof(head));
        etot=0;
    }
    void add(int a,int b) 
    {
        int key=a%S;
        for (int t=head[key];t;t=last[t])
        if (dest[t][0]==a) return;
        etot++;
        dest[etot][0]=a;
        dest[etot][1]=b;
        last[etot]=head[key];
        head[key]=etot;
    }
    int find(int a) 
    {
        int key=a%S;
        for (int t=head[key];t;t=last[t])
        if (dest[t][0]==a) return dest[t][1];
        return -1;
    }
}hash;
void exgcd(int a,int b,int &d,int &x,int &y)
{
    if (b==0) {d=a;x=1;y=0;}
    else 
    {
        int xx,yy;
        exgcd(b,a%b,d,xx,yy);
        //xx*b+yy*(a-(a/b)*b)=d
        //a*yy+b*(xx-(a/b)*yy)=d
        x=yy;
        y=xx-(a/b)*yy;
    }
}
int inverse(int a,int m)
{
    int d,x,y;
    exgcd(a,m,d,x,y);
    assert(d==1);
    return (x%m+m)%m;
}
int ind(int a, int b, int m) 
{
    int c=(int)ceil(sqrt(m)+1);
//  hash.init();
    map<int,int> mp;
    int cur=1;
    for(int i=0;i<c;i++,cur=(1LL*cur*a)%m) 
    {
//      hash.add(cur,i);
        if (mp.count(cur)==0) mp[cur]=i;
        if (b==cur) return i;
    }
    int base=inverse(cur,m);    //a^{-c}
    cur=1LL*b*base%m;   //b*a^{-c}
    for (int i=c;i<=m;i+=c,cur=(1LL*cur*base)%m) 
    {
//      int j=hash.find(cur);
        map<int,int>::iterator it=mp.find(cur);
        if (it!=mp.end()) return it->second+i;
    }
    assert(false);
    return -1;
}
int main() 
{
    int a,b,m;
    scanf("%d%d%d",&a,&b,&m);
    int x=ind(a,b,m);
    printf("%d\n",x);
    return 0;
}