BZOJ_P2242&Codevs_P1565[SDOI2011]计算器(快速幂+扩展欧几里得+BSGS)

BZOJ传送门 Codevs传送门
Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 2302 Solved: 910
[Submit][Status][Discuss]
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

HINT
Source
第一轮day1

参考hzwer学长
感谢zky学长
ps:旁边的神犇:”辣鸡的模板题。”

#include<cstdio>
#include<cmath>
#include<map>
#include<iostream>
using namespace std;
inline int in(){
    int x=0;char ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
inline long long pow(long long a,int b,int p){
    long long res=1;
    while(b){
        if(b&1) res=res*a%p;
        a=a*a%p;b>>=1;
    }
    return res;
}
inline int gcd(int a,int b){return b==0?a:gcd(b,a%b);}
inline void exgcd(int a,int b,int &x,int &y){
    if(!b){x=1,y=0;return;}
    exgcd(b,a%b,x,y);
    int t=x;x=y,y=t-a/b*y;
}
inline void solve1(int t){
    int a,b,p;long long ans;
    while(t--){
        a=in(),b=in(),p=in();ans=pow(a,b,p);
        printf("%lld\n",ans);
    }
}
inline void solve2(int t){
    int a,b,p,x,y,d;
    while(t--){
        a=in(),b=in(),p=in();
        p=-p;d=gcd(a,p);
        if(b%d){puts("Orz, I cannot find x!");continue;}
        a/=d,b/=d,p/=d;exgcd(a,p,x,y);
        x=(long long)x*b%p;while(x<0) x+=p;
        printf("%d\n",x);
    }
}
inline void solve3(int t){
    int a,b,p;map<int,int> mp;bool f;
    while(t--){
        a=in(),b=in(),p=in();
        a%=p;
        if(!a&&!b){puts("1");continue;}
        if(!a){puts("Orz, I cannot find x!");continue;}
        mp.clear();
        long long m=ceil(sqrt(p)),d=1;mp[1]=m+1;
        for(long long i=1;i<m;i++){
            d=d*a%p;
            if(!mp[d]) mp[d]=i;
        }
        long long tmp=pow(a,p-m-1,p),ine=1;f=true;
        for(long long k=0;k<m&&f;k++){
            int i=mp[b*ine%p];
            if(i){
                if(i==m+1) i=0;
                printf("%lld\n",k*m+i);
                f=false;break;
            }
            ine=ine*tmp%p;
        }
        if(f) puts("Orz, I cannot find x!");
    }
}
int main(){
    int t,k;
    scanf("%d%d",&t,&k);
    switch(k){
        case 1:solve1(t);break;
        case 2:solve2(t);break;
        case 3:solve3(t);break;
    }
    return 0;
}