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，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;
}