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

第一问快速幂求得。

第二问求axΞ b(mod n)，转化为ax=ny+b，转化为ax+ny=b，利用拓展欧几里得算法求出ax+ny=gcd(a,n)，如果b不是gcd的倍数则无解否则为x/gcd*b。

第三问求a^{x Ξ}b(mod n)，BSGS算法。我们需要验证0..n-1内的数。分块，设每块大小为m，求出0..m-1内的aⁱ % n保存为ei，对于m..2m-1内的数，我们只需要验证是否存在有a^m *ei=b(mod n)，即判断是否存在ei=a^-m *b (mod n)，这样用一个hash表存一下ei然后求一下在模n下a^m的逆元就可以了。

时间复杂度为O((m+n/m)logm)，当m取n^½的时候复杂度较优为O(n^½logn)

【代码】

 #include<map>
 #include<cmath>
 #include<cstdio>
 using namespace std;

 typedef long long LL;
 LL a,b,c,T,k;

 LL pow(LL x,LL p,LL MOD) {
     LL tmp=x,ans=;
     while(p) {
         if(p&) ans=(ans*tmp)%MOD;
         tmp=(tmp*tmp)%MOD;
         p>>=;
     }
     return ans;
 }
 void gcd(LL a,LL b,LL& d,LL& x,LL& y) {
     if(!b) d=a,x=,y=;
     else gcd(b,a%b,d,y,x),y-=x*(a/b);
 }
 LL inv(LL a,LL n) {
     LL d,x,y;
     gcd(a,n,d,x,y);
     return d==? (x+n)%n:-;
 }
 LL log_mod(LL a,LL b,LL n) {
     LL m,v,e=,i;
     m=sqrt(n+0.5);
     v=inv(pow(a,m,n),n);
     map<LL,LL> mp;
     mp[]=;
     for(LL i=;i<m;i++) {
         e=(e*a)%n;
         if(!mp.count(e)) mp[e]=i;
     }
     for(LL i=;i<m;i++) {
         if(mp.count(b)) return i*m+mp[b];
         b=(b*v)%n;
     }
     return -;
 }

 int main() {
     scanf("%lld%lld",&T,&k);
     while(T--) {
         scanf("%lld%lld%lld",&a,&b,&c);
         if(k==) {
             printf("%lld\n",pow(a,b,c));
         } else
         if(k==) {
             LL d,x,y;
             gcd(a,c,d,x,y);
             if(b%d) puts("Orz, I cannot find x!");
             else {
                 LL ans=((x*b/d)%c+c)%c;
                 printf("%lld\n",ans);
             }
         } else {
             LL ans=log_mod(a,b,c);
             if(ans==-) puts("Orz, I cannot find x!");
             else printf("%lld\n",ans);
         }
     }
     return ;
 }