BZOJ 2242: [SDOI2011]计算器

2242: [SDOI2011]计算器

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

快速幂，扩展欧几里得，bsgs三合一

注意不要爆int，用long long

这个题好像不忽略行尾控空格回车，注意不要格式错误

2017.6.9  更改bsgs代码为减法版y^(i*m-j)=z(mod p)

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
inline void write(long long x)
{
    if(x==0){putchar('0');return;}
    int temp[30];temp[0]=0;if(x<0)putchar('-'),x=-x;
    while(x){temp[++temp[0]]=x%10;x/=10;}
    while(temp[0]){putchar(temp[temp[0]--]+'0');}
}
int opt,n;
inline long long qpow(long long x,int y,int z)
{
    long long ans=1;
    while(y)
    {
        if(y&1)ans=ans*x%z;
        x=x*x%z;
        y>>=1;
    }
    return ans;
}
void qm()
{
    int x,y,z;
    for(int i=1;i<=n;i++)
    {
        x=read();y=read();z=read();
        write(qpow(x,y,z));puts("");
    }
}
int u,q;
inline ll exgcd(ll a,ll b)
{
    if(b==0){u=1;q=0;return a;}
    ll temp=exgcd(b,a%b);
    ll t=u;u=q;q=t-a/b*q;
    return temp;
}
void ko()
{
    ll x,y,z;
    for(int i=1;i<=n;i++)
    {
        x=read();y=read();z=read();y%=z;
        int t=exgcd(x,z);
        if(y%t){puts("Orz, I cannot find x!");continue;}
        y/=t;x/=t;z/=t;u=u*y%z;
        while(u<0)u+=z;
        write(u);puts("");
    }
}
map<int,int>mp;
inline void bsgs(int y,int z,int p)
{
	y%=p;
	if(!y&&!z){puts("1");return ;}
	if(!y){puts("Orz, I cannot find x!");return;}	
	mp.clear();int m=ceil(sqrt(p));ll tmp=z%p;
	for(int i=1;i<=m;i++)
	{tmp=tmp*y%p;mp[tmp]=i;}
	tmp=qpow(y,m,p);ll ine=1;
	for(int i=1;i<=m;i++)
	{
		ine=ine*tmp%p;
		if(mp[ine])
		{
			printf("%d\n",((i*m-mp[ine])%p+p)%p);
			return ;
		}		
	}
	puts("Orz, I cannot find x!");
}
void b()
{
    ll x,y,z;
    for(int i=1;i<=n;i++)
    {
        x=read();y=read();z=read();
        bsgs(x,y,z);
    }
}
int main()
{
    int x,y,z;
    n=read();opt=read();
    switch(opt)
    {
        case 1:qm();break;
        case 2:ko();break;
        case 3:b();break;
    }
    return 0;
}