[bzoj2242]计算器 [三个板子]

[bzoj2242]计算器

---

三个板子，以下是需要注意的地方

1. qpow的板子不用说了，不要把取模算错就是了。
2. 解同余方程的时候最后小心乘爆
3. bsgs先取下模然后特判以下0

• 代码

#include<bits/stdc++.h>
using namespace std;
int T,K;
typedef long long ll;

int qpow(int a,int b,int p){
    int res=1;
    while(b){
        if(b&1)res=(long long)res*a%p;
        b>>=1;
        a=(long long)a*a%p;
    }
    return res;
}

void exgcd(int a,int b,long long &d,long long &x,long long &y){
    if(b==0){
        d=a,x=1,y=0;
    }else{
        exgcd(b,a%b,d,y,x);
        y-=(a/b)*x;
    }
}

int inv(int x,int p){
    return qpow(x,p-2,p);
}

map<int,int>mp;

void bsgs(int y,int z,int p){
    z%=p,y%=p;
    if (y==0&&z==0) {printf("1\n");return;}
    if (y==0&&z!=0) {printf("Orz, I cannot find x!\n");return;}
    mp.clear();
    int sqr=sqrt(p);
    int _inv=inv(qpow(y,sqr,p),p);
    int tmp=1;
    for(int i=0;i<sqr;i++){
        if(!mp[tmp])mp[tmp]=i;
        tmp=(long long)tmp*y%p;
    }
    int pres=z;
    for(int i=0;i<=p;i+=sqr){
        if(mp[pres]){
            printf("%d\n",i+mp[pres]);
            return;
        }
        pres=(long long)pres*_inv%p;
    }
    printf("Orz, I cannot find x!\n");
}

int main()
{
    scanf("%d%d",&T,&K);
    while(T--){
        int y,z,p;
        scanf("%d%d%d",&y,&z,&p);
        if(K==1) printf("%d\n",qpow(y,z,p));
        else if(K==2){
            long long g=0,x1=0,x2=0; 
            exgcd(y,p,g,x1,x2);
            if(z%g){printf("Orz, I cannot find x!\n");continue;}
            x1%=p/g;x1*=z/g;x1%=p/g;
            if(x1<0)x1+=p/g;
            printf("%d\n",x1);
        }else
            bsgs(y,z,p);
    }
}