【BZOJ】【2242】【SDOI2011】计算器

快速幂/扩展欧几里得/BSGS

　　经典好例题！！

　　三个问题三种算法……

　　算法：白书（算法竞赛入门经典——训练指南）全有……

 /**************************************************************
     Problem: 2242
     User: Tunix
     Language: C++
     Result: Accepted
     Time:1824 ms
     Memory:2476 kb
 ****************************************************************/

 //BZOJ 2242
 #include<map>
 #include<cmath>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #define rep(i,n) for(int i=0;i<n;++i)
 #define F(i,j,n) for(int i=j;i<=n;++i)
 #define D(i,j,n) for(int i=j;i>=n;--i)
 using namespace std;

 int getint(){
     int v=,sign=; char ch=getchar();
     while(ch<''||ch>'') {if (ch=='-') sign=-; ch=getchar();}
     while(ch>=''&&ch<='') {v=v*+ch-''; ch=getchar();}
     return v*=sign;
 }
 /*******************tamplate********************/
 typedef long long LL;
 LL x,y,z,P;
 LL pow(LL a,LL b,LL P){
     LL r=,base=a;
     while(b){
         if (b&) r=r*base%P;
         base=base*base%P;
         b>>=;
     }
     return r;
 }
 void exgcd(LL a,LL b,LL &d,LL &x,LL &y){
     if (!b) {d=a; x=; y=; return;}
     else{exgcd(b,a%b,d,y,x); y-=x*(a/b);}
 }
 LL log_mod(LL a,LL b,LL P){
     LL m,v,e=,i;
     m=ceil(sqrt(P+0.5));//这里需要用ceil……？
     v=pow(a,P-m-,P);//inv
     map<LL,LL> x;
     x[]=;
     for(int i=;i<m;++i){
         e=e*a%P;
         if (!x.count(e)) x[e]=i;
     }
     rep(i,m){
         if (x.count(b)) return (LL)i*m+x[b];
         b=b*v%P;
     }
     return -;
 }
 int main(){
     int T=getint(),K=getint();
     while(T--){
         y=getint(); z=getint(); P=getint();
         if (K==) printf("%lld\n",pow(y,z,P));
         if (K==){
             LL d=,X=,Y=;
             exgcd(y,P,d,X,Y);
             if(z%d){ printf("Orz, I cannot find x!\n"); continue; }
             X*=z/d;
             LL T=X*d/P;
             X-=T*P/d;
             if (X<) X+=P/d;
             printf("%lld\n",X);
         }
         if (K==){
             y%=P; z%=P;
             if(!y && !z) {printf("1\n"); continue;}
             if(!y) { printf("Orz, I cannot find x!\n"); continue; }
             LL ans=log_mod(y,z,P);
             if (ans==-){ printf("Orz, I cannot find x!\n"); continue; }
             printf("%lld\n",ans);
         }
     }
     return ;
 }