hdu 2815 Mod Tree pku 3243 Clever Y

扩展Baby Step Giant Step

Baby Step Giant Step： A^X=B(mod C); C为素数。

如果C没有限制就变成扩展Baby Step Giant Step。

参照大牛博客：http://hi.baidu.com/aekdycoin/item/236937318413c680c2cf29d4

实在是太弱了，没用二分，写了一个hash竟然超时，最后还是copy了一个hash。

# include<stdio.h>
# include<string.h>
# include<stdlib.h>
# include<math.h>
# define INF 65535
int X,Y;
//int Hash[INF],val[INF];
struct hash{
    int a,b,next;
}Hash[INF << 1];
int flg[INF + 66];
int top,idx;
void ins(int a,int b){
    int k = b & INF;
    if(flg[k] != idx){
        flg[k] = idx;
        Hash[k].next = -1;
        Hash[k].a = a;
        Hash[k].b = b;
        return ;
    }
    while(Hash[k].next != -1){
        if(Hash[k].b == b) return ;
        k = Hash[k].next;
    }
    Hash[k].next = ++ top;
    Hash[top].next = -1;
    Hash[top].a = a;
    Hash[top].b = b;
}
int find(int b){
    int k = b & INF;
    if(flg[k] != idx) return -1;
    while(k != -1){
        if(Hash[k].b == b) return Hash[k].a;
        k = Hash[k].next;
    }
    return -1;
}

int gcd(int a,int b)
{
    int tmp;
    while(a%b)
    {
        tmp=a%b;
        a=b;
        b=tmp;
    }
    return b;
}
void extend_gcd(int a,int b)
{
    int X1,Y1;
    if(b==0)
    {
        X=1;
        Y=0;
        return;
    }
    extend_gcd(b,a%b);
    X1=Y;
    Y1=X-Y*(a/b);
    X=X1;
    Y=Y1;
    
}
int Inval(int a,int b,int n) // 求解方程a*x+b*y=n;
{
    int e;
    extend_gcd(a,b);
    e=(__int64)X*n%b;
    return e<0?e+b:e;
}
int pow_mod(int A,int m,int C)
{
    int mm,i;
    mm=1;
    for(i=1;i<=m;i++)
        mm=(__int64)mm*A%C;
    return mm;
}
/*void insert(int x,int i,int m)
{
int pos;
pos=x%m;
while(Hash[pos]!=-1)
{
pos++;
if(pos==m) pos=0;
}
Hash[pos]=i;
val[pos]=x;
}
int find(int x,int m)
{
int pos,i,pos1;
pos=x%m;
for(i=0;i<m;i++)
{
pos1=pos+i;
if(pos1>=m) pos1-=m;
if(Hash[pos1]==-1) break;
if(val[pos1]==x) break;
}
if(i==m) return -1;
return Hash[pos1];
}*/

int babystep(int A,int C,int B)
{
    int i,d,tmp,m,w,mm,ans;
    __int64 D;
    top = INF; ++ idx; 
    d=0;
    D=1;
    ans=1;
    for(i=0;i<=100;i++)
    {
        if(ans==B) return i;
        ans=(__int64)ans*A%C;
    }
    tmp=gcd(A,C);
    while(tmp!=1)
    {
        if(B%tmp) return -1;
        ++d;
        C/=tmp;
        B/=tmp;
        D=D*A/tmp%C;
        tmp=gcd(A,C);
    }
    m=ceil(sqrt((double)C));
    ans=1;
    //memset(Hash,-1,sizeof(Hash));
    for(i=0;i<=m;i++)
    {
        //insert(ans,i,m+1);//把A^i%C插入hash表
        ins(i,ans);
        ans=(__int64)ans*A%C;
    }
    mm=pow_mod(A,m,C); //A^m%C
    for(i=0;i<=m;i++) // mm^i
    {
        tmp=Inval((int)D,C,B);
        if(tmp>=0 && (w=find(tmp))!=-1) return i*m+w+d;
        D=D*mm%C;
    }
    return -1;
}
int main()
{
    int K,P,N,ans;
    while(scanf("%d%d%d",&K,&P,&N)!=EOF)
    {
        //if(K==0 && P==0 && N==0) break;
        if(N>=P) {printf("Orz,I can’t find D!\n");continue;}
        if(P==1)
        {
            if(N==0) printf("0\n");
            else printf("Orz,I can’t find D!\n");
            continue;
        }
        ans=babystep(K,P,N);
        if(ans==-1) printf("Orz,I can’t find D!\n");
        //if(ans==-1) printf("No Solution\n");
        else printf("%d\n",ans);
    }
    return 0;
}