2242: [SDOI2011]计算器
Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 2337 Solved: 930
[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
题解:这是一道练习基本数论的题目。
询问1:快速幂,不解释,但是要注意中间过程,不要爆了
询问2:扩展欧几里德算法
xy mod p=z 先用扩展欧几里得求xy mod p=1,然后答案再乘z,因为要求的是最小非负整数,所以最后的结果要一直加p直到为正。
询问3:bsgs算法,之前博客里有一篇文章详细解释过,不再解释。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
using namespace std;
int i,j,n,a,b,c,tt,k,m;
long long x,y;
map<long long,int> mp;
long long quickpow(int x)
{
long long ans=1; long long l=a;
while (x>0)
{
if (x&1)
ans=(ans%c*l%c)%c;
x=x>>1;
l=(l%c*l%c)%c;
}
return ans;
}
void work1()
{
printf("%lld\n",quickpow(b));
}
long long gcd(long long x,long long y)
{
long long r;
while (y!=0)
{
r=x%y;
x=y;
y=r;
}
return x;
}
void exgcd(long long a,long long b)
{
if (b==0)
{
x=1;y=0;
return;
}
else
{
exgcd(b,a%b);
long long t=y;
y=x-(a/b)*y;
x=t;
}
}
void work2()
{
int g=gcd(a,c);
if (b%(gcd(a,c))!=0)
{
printf("Orz, I cannot find x!\n");
return;
}
a/=g; b/=g; c/=g;
exgcd(a,c);
x=(long long)(x%c*b%c)%c;
while (x<0) x+=c;
printf("%lld\n",x);
}
void work3()
{
mp.clear();
if (a%c==0)
{
printf("Orz, I cannot find x!\n"); return;
}
m=ceil(sqrt(c));
long long ans=b; mp[b%c]=0;
for (int i=1;i<=m;i++)
{
ans=(ans%c*a%c)%c;
if (!mp[ans]) mp[ans]=i;
}
long long l=quickpow(m); ans=1;
for (int i=1;i<=m;i++)
{
ans=(ans%c*l%c)%c;
if (mp[ans]!=0)
{
int mm=i*m-mp[ans];
printf("%d\n",(mm%c+c)%c);
return;
}
}
printf("Orz, I cannot find x!\n");
}
int main()
{
scanf("%d%d",&tt,&k);
for (i=1;i<=tt;i++)
{
scanf("%d%d%d",&a,&b,&c);
if(k==1) work1();
if (k==2) work2();
if (k==3) work3();
}
}