bzoj2242

2242: [SDOI2011]计算器

Time Limit: 10 Sec   Memory Limit: 512 MB
Submit: 3179   Solved: 1248
[ 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

[ Submit][ Status][ Discuss]

这是一道题，也是3道题。。。前两道SB题,最后一道要求y^x=z(mod p)的x值

由于费马小定理,答案若存在一定不会>p-1

于是我们可以分块乱搞 

y^x=y^(k*sqrt(p)+i)=z (0<=i<sqrt(x))

即 y^i=z/(y^(k*sqrt(p))=z*inv(y^(k*sqrt(p)))=z*y^(k*sqrt(p)*(p-2)%(p-1))

哈希表乱搞即可

这个算法的学名叫大步小步算法--By mcfx

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<tr1/unordered_map>
using namespace std;
tr1::unordered_map<int,int>st;
int qmod(int a,int b,int p){
	int tmp=a,ans=1;
	for(int i=0;b>>i;i++,tmp=1ll*tmp*tmp%p)
		if((b>>i)&1)ans=1ll*ans*tmp%p;
	return ans;
}
void exgcd(int a,int b,int& d,int &x,int &y){
	if(b==0){
		x=1;y=0;d=a;
	} else exgcd(b,a%b,d,y,x),y-=a/b*x;
}
int get(int y,int z,int p){
	int d,x,m;
//	printf("[%d,%d]",y,-p);
	exgcd(y,p,d,x,m);
	if(z%d!=0)return -1;
	while(x<0)x+=p;
	return 1ll*x*z%p;
}
int get2(int y,int z,int p){
	if(y%p==0)return z==0?1:-1;
	int mxs=sqrt(p)+1;
	int f=qmod(y,mxs,p);
	st.clear();
	for(int i=0,j=1;i<mxs;++i,j=1ll*j*y%p)
		if(!st.count(j))st[j]=i;
	for(int i=0;i<=p;i+=mxs){
		int ans=1ll*z*qmod(y,1ll*i*(p-2)%(p-1),p)%p;
		if(st.count(ans))return st[ans]+i;
	}
	return -1;
}
void print(int x){
	x==-1?printf("Orz, I cannot find x!\n"):printf("%d\n",x);
}
int main(){
	int y,z,p,t,k,x;
	scanf("%d%d",&t,&k);
	while(t--){
		scanf("%d%d%d",&y,&z,&p);
		if(k==1)printf("%d\n",qmod(y,z,p));
		else if(k==2)print(get(y,z,p));
		else if(k==3)print(get2(y,z,p));
	}
}