bzoj2242 [SDOI2011]计算器(BSGS+Hash表+乘法逆元)

操作3是bsgs裸题。
注意特判a是p的倍数的情况。
没想到死在了操作2上了【再见】

#include 
#include 
#include 
#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100100
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(S==T) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int mod,m,h[N],num=0;
struct Hash_table{
    int next,key,val;
}data[N];
inline void ins(int key,int val){
    int x=key%N;
    for(int i=h[x];i;i=data[i].next){
        if(data[i].key==key) return;
    }data[++num].next=h[x];h[x]=num;data[num].key=key;data[num].val=val;
}
inline int hs(int key){
    int x=key%N;
    for(int i=h[x];i;i=data[i].next){
        if(data[i].key==key) return data[i].val;
    }return -1;
}
inline int ksm(ll x,int k){
    ll res=1;for(;k;k>>=1,x=x*x%mod) if(k&1) res=res*x%mod;return res;
}
inline int BSGS(int a,int b){
    if(b==1) return 0;
    if(a==0) return b==0?1:-1;
    m=ceil(sqrt(mod));memset(h,0,sizeof(h));num=0;
    int e=ksm(a,mod-2);ll x=b;
    for(int i=0;i1;
    for(int i=0;i<=m;++i){
        int j=hs(x);
        if(~j) return i*m+j;x=x*e%mod;
    }return -1;
}
int main(){
//  freopen("a.in","r",stdin);
    int tst=read(),type=read();
    if(type==1){
        while(tst--){
            int x=read(),y=read();mod=read();
            printf("%d\n",ksm(x,y));
        }
    }if(type==2){
        while(tst--){
            int x=read(),y=read();mod=read();x%=mod;y%=mod;
            if(!x&&y) puts("Orz, I cannot find x!");
            else printf("%lld\n",(ll)y*ksm(x,mod-2)%mod);
        }
    }if(type==3){
        while(tst--){
            int a=read(),b=read();mod=read();b%=mod;a%=mod;
            int res=BSGS(a,b);
            if(res==-1) puts("Orz, I cannot find x!");
            else printf("%d\n",res);
        }
    }return 0;
}

你可能感兴趣的:(bzoj,乘法逆元,BSGS)