Hdu 2815 Mod Tree —— exbsgs模板

This way

题意：

告诉你k,p,n，问你$K^x\%P=N$

题解：

被坑了…他这个不是同余N而是等于N，也就是说N>=P时无解。
证明就不详细说了，大致就是利用同余定理，将a与p约到gcd为1的时候，最终的式子是
$$a^{x-k}\ast \frac{a^k}{\sum _{i=1}^{k}gc{d}_i}=\frac{b}{\sum _{i=1}^{k}gcd}\%\frac{mod}{\sum _{i=1}^{k}gc{d}_i}$$
那么将a^k部分移到右边，接下来就是和bsgs相同的了

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll mod;
const ll inf=1e10; 
ll qpow(ll a,ll b){ll ans=1;for(;b;b>>=1,a=a*a%mod)if(b&1)ans=ans*a%mod;return ans;}
map<ll,int>mp;
void ex_gcd(ll a,ll b,ll& x,ll& y){
    if(!b){ 
        x=1,y=0;
    }
    else {
        ex_gcd(b,a%b,y,x);
        y=y-a/b*x;
    }
}
int main()
{
    ll a,b,n;
    while(scanf("%lld%lld%lld",&a,&mod,&b)!=EOF){
        if(b>=mod){//同余时去掉
            printf("Orz,I can’t find D!\n");
            continue;
        }
        a%=mod,b%=mod;
        if(b==1){
            printf("0\n");
            continue;
        }
        mp.clear();
        ll g=__gcd(a,mod),ret=1,k=0;
        while(g!=1){//取出所有gcd
            if(b%g){
                printf("Orz,I can’t find D!\n");
                k=-1;
                break;
            }
            k++;
            b/=g,mod/=g;
            ret=ret*(a/g)%mod;
            g=__gcd(a,mod);
            if(b==ret){
                printf("%lld\n",k);
                k=-1;
                break;
            }
        }
        if(k==-1)continue;
        ll inv,y;
        ex_gcd(ret,mod,inv,y);
        inv=(inv%mod+mod)%mod;
        b=b*inv%mod,a%=mod; //将gcd乘到b上
        int x=min(2500,(int)sqrt(mod)+1);
        ll v=1;
        ll ans=inf;
        for(int i=0;i<=x;i++,v=v*a%mod){//a的i次
            if(mp.count(v))break;
            mp[v]=i;
            if(v==b)ans=min(ans,1ll*i+k);
        }
        v=b;
        ret=qpow(a,x);
        ex_gcd(ret,mod,inv,y);
        inv=(inv%mod+mod)%mod;
        ret=inv;
        ll top=mod/x+1;
        for(int i=0;i<=top;i++){//a的xi次
            if(mp.count(v))ans=min(ans,1ll*i*x+mp[v]+k);
            if(v==1&&i!=0)ans=min(ans,1ll*i*x+k);
            if(i*x+k>=ans)break;
            v=v*ret%mod;
        }
        if(ans==inf)printf("Orz,I can’t find D!\n");
        else printf("%lld\n",ans);
    }
    return 0;
}