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,P为质数,1<=T<=10。
Sample Output
【样例输出1】
2
1
2
【样例输出2】
2
1
0
首先,任务一很简单,是一个快速幂,代码如下:
long long quick(long long a,long long b,long long p)
{
long long ans=1;
while(b)
{
if(b&1)ans=ans*a%p;
a=a*a%p;
b/=2;
}
return ans;
}
其次,任务二是一个exgcd,就不多说了,代码块如下:
int exgcd(long long a,long long b,int &x,int &y)
{
if(!b){x=1; y=0; return a;}
int d=exgcd(b,a%b,x,y);
int temp=y;
y=x-a/b*y;
x=temp;
return d;
}
最后,任务三就是一个bsgs,关于bsgs的详解,请查看BSGS这篇文章,然后bsgs部分的代码如下:
int bsgs(long long p,long long b,long long n)
{
b%=p;
int t=ceil(sqrt(p));
for(int i=0;i<t;i++)
{
ha[1ll*n*quick(b,i,p)%p]=i;
}
b=quick(b,t,p);
if(!b) return n?-1:1;
for(int i=0;i<=t;i++)
{
int val=quick(b,i,p);
int j=(ha.find(val)==ha.end())?-1:ha[val];
if(i*t-j>=0&&j>=0)return i*t-j;
}
return -1;
}
最后附上AC代码:
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<queue>
#include<map>
#include<string>
#include<cstring>
#include<vector>
#include<stack>
#include<set>
#define int long long
using namespace std;
const int MAXN=1e6+10;
map<int ,int >ha;
long long quick(long long a,long long b,long long p)
{
int ans=1;
while(b)
{
if(b&1)ans=1ll*ans*a%p;
a=1ll*a*a%p;
b/=2;
}
return ans;
}
int bsgs(long long p,long long b,long long n)
{
b%=p;
int t=ceil(sqrt(p));
for(int i=0;i<t;i++)
{
ha[1ll*n*quick(b,i,p)%p]=i;
}
b=quick(b,t,p);
if(!b) return n?-1:1;
for(int i=0;i<=t;i++)
{
int val=quick(b,i,p);
int j=(ha.find(val)==ha.end())?-1:ha[val];
if(i*t-j>=0&&j>=0)return i*t-j;
}
return -1;
}
int exgcd(long long a,long long b,int &x,int &y)
{
if(!b)
{
x=1;y=0;return a;
}
int d=exgcd(b,a%b,x,y);
int temp=y;
y=x-a/b*y;
x=temp;
return d;
}
signed main()
{
long long y,z,p;
int t,q;
cin>>q>>t;
while(q--)
{
cin>>y>>z>>p;
if(t==3)
{
ha.clear();
int a=bsgs(p,y,z);
if(~a)cout<<a<<endl;
else cout<<"Orz, I cannot find x!"<<endl;
}
else if(t==1)
{
cout<<quick(y,z,p)<<endl;
}
else if(t==2)
{
int m,x;
int a=exgcd(y,p,x,m);
if(z%a)cout<<"Orz, I cannot find x!"<<endl;
else{
int k=p/a;
cout<<(x*z/a%k+k)%k<<endl;
}
}
}
return 0;
}