【BZOJ 2242】[SDOI2011]计算器

Description

你被要求设计一个计算器完成以下三项任务：

1、给定y,z,p,计算Y^Z Mod P 的值；

2、给定y,z,p，计算满足xy≡ Z ( mod P )的最小非负整数；

3、给定y,z,p，计算满足Y^x ≡ Z ( mod P)的最小非负整数。

Input

输入包含多组数据。

第一行包含两个正整数T,K分别表示数据组数和询问类型（对于一个测试点内的所有数据，询问类型相同）。

以下行每行包含三个正整数y,z,p，描述一个询问。

Output

对于每个询问，输出一行答案。对于询问类型2和3，如果不存在满足条件的，则输出“Orz, I cannot find x!”，注意逗号与“I”之间有一个空格。

Sample Input

【样例输入1】
3 1
2 1 3
2 2 3
2 3 3
【样例输入2】
3 2
2 1 3
2 2 3
2 3 3
【数据规模和约定】
对于100%的数据，1<=y,z,p<=10^9，为质数，1<=T<=10。

Sample Output

【样例输出1】
2
1
2
【样例输出2】
2
1
0

 

第一问第二问略过，第三问BSGS

-------------------------------------------------------------叫我分割线-----------------------------------------------------------

什么是BSGS呢？即baby-step-giant-step，翻译成中文就是小步大步法，用于解决类似x^y=z(mod  p) 求最小的y这样的问题（也许还能干别的，但本人弱渣，并不知道）

对于上面那个题目的推导

 

有点凌乱，等我想明白的再补

 #include<cstdio>
 #define ll long long
 #include<map>
 #include<cmath>
 using namespace std;
 int T,k;
 ll pow(ll x,int y,int p){
     ll ans=;
     while(y>){
         if (y&==) ans=(ans*x)%p;
         y=y>>;
         x=(x*x)%p;
     }
     return ans;
 }

 int gcd(int x,int y){
     if (x%y==) return y;
     return gcd(y,x%y);
 }

 void exgcd(int a,int b,int &x,int &y){
     if (b==){x=,y=;return;}
     exgcd(b,a%b,x,y);
     int t=x;x=y;y=t-(a/b)*y;
 }

 void solve2(int a,int z,int b){
     int tmp=gcd(a,b),x,y;
     if (z%tmp){printf("Orz, I cannot find x!\n");return;}
     exgcd(a,b,x,y);
     x=((ll)x*(z/tmp))%b;
     while (x>) x-=b/tmp;
     while (x<) x+=b/tmp;
     printf("%d\n",x);
 }

 map<int,int> mp;
 void solve3(int y,int z,int p){
     y%=p;
     if (!y&&!z) {printf("1\n");return;}
     if (!y){printf("Orz, I cannot find x!\n");return;}
     mp.clear();
     ll m=ceil(sqrt(p)),t=;
         mp[]=m+;//y^0==1;
     for (int i=;i<m;i++){
         t=t*y%p;
         if (!mp[t]) mp[t]=i;
     }
     ll tmp=pow(y,p--m,p),ine=;
     for (int k=;k<m;k++){
         int i=mp[z*ine%p];
         if (i){
             if (i==m+)i=;
             printf("%d\n",k*m+i);
             return;
         }
         ine=ine*tmp%p;
     }
     printf("Orz, I cannot find x!\n");
 }

 int main(){
     scanf("%d%d",&T,&k);
     while (T--){
         int y,z,p;
         scanf("%d%d%d",&y,&z,&p);
         if (k==) printf("%lld\n",pow(y,z,p));
         if (k==) solve2(y,z,p);
         if (k==) solve3(y,z,p);
     }
 }