BZOJ 2242 SDOI2011 计算器 快速幂+扩展欧几里得+BSGS

题目大意：……简洁明了自己看

第一问快速幂

第二问扩展欧几里得

第三问BSGS

顺便一开始没看到p是质数0.0 去弄了EXBSGS的模板0.0 懒得改了

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 1001001
using namespace std;
typedef long long ll;  
typedef pair<ll,ll> abcd;  
ll A,B,C,D,hash_table[M],val[M],tim[M],tot;  
int Hash(ll x)  
{  
    int pos=x%M;  
    while(1)  
    {  
        if(tim[pos]!=tot)  
            tim[pos]=tot,hash_table[pos]=-1,val[pos]=0x3f3f3f3f;  
        if(hash_table[pos]==-1||hash_table[pos]==x)  
            return hash_table[pos]=x,pos;  
        else  
            ++pos,pos%=M;  
    }  
}  
int Get_Hash(ll x)  
{  
    int pos=x%M;  
    while(1)  
    {  
        if(tim[pos]!=tot)  
            tim[pos]=tot,hash_table[pos]=-1;  
        if(hash_table[pos]==-1)  
            return -1;  
        if(hash_table[pos]==x)  
            return pos;  
        else  
            ++pos,pos%=M;  
    }  
}  
ll GCD(ll x,ll y)  
{  
    return y?GCD(y,x%y):x;  
}  
abcd EXGCD(ll x,ll y)  
{  
    if(!y) return abcd(1,0);  
    abcd temp=EXGCD(y,x%y);  
    return abcd(temp.second,temp.first-x/y*temp.second);  
}  
ll Inverse(ll x)  
{  
    ll temp=EXGCD(x,C).first;  
    return (temp%C+C)%C;  
}  
ll EXBSGS()  
{  
    ll i,m,cnt=0,temp,base=1;  
    int pos;  
    B%=C;  
    for(i=0,temp=1%C;i<=50;i++,temp*=A,temp%=C)  
        if(temp==B)  
            return i;  
    D=1;  
    while(temp=GCD(A,C),temp!=1)  
    {  
        if(B%temp)  
            return -1;  
        ++cnt;  
        B/=temp;  
        C/=temp;  
        D*=A/temp;  
        D%=C;  
    }  
    B*=Inverse(D);B%=C;  
    m=(ll)ceil(sqrt(C)+1e-5);  
    ++tot;  
    for(i=0,temp=1%C;i<m;i++,temp*=A,temp%=C)  
        pos=Hash(temp),val[pos]=min(val[pos],i);  
    for(i=1,base=1%C;i<=m;i++,base*=A,base%=C);  
    for(i=0,D=1%C;i<m;i++,D*=base,D%=C)  
    {  
        temp=EXGCD(D,C).first*B;  
        temp=(temp%C+C)%C;  
        pos=Get_Hash(temp);  
        if(~pos)  
            return i*m+val[pos]+cnt;  
    }  
    return -1;  
}
ll KSM()
{
	ll re=1;
	while(B)
	{
		if(B&1)re*=A,re%=C;
		A*=A,A%=C;
		B>>=1;
	}
	return re;
}
ll EXGCD()
{
	if(A%C==0&&B%C!=0)
		return -1;
	return (EXGCD(A,C).first%C*B%C+C)%C;
}
int main()
{
	int T,k;
	ll ans;
	for(cin>>T>>k;T;T--)
	{
		cin>>A>>B>>C;
		switch(k)
		{
			case 1:ans=KSM();break;
			case 2:ans=EXGCD();break;
			case 3:ans=EXBSGS();break;
		}
		if(ans==-1)
			puts("Orz, I cannot find x!");
		else
			cout<<ans<<endl;
	}
}