【BSGS】2019牛客暑期多校训练营（第五场）- C - generator 2

题目链接https://ac.nowcoder.com/acm/contest/885/C

---

题意

给出式子$x_i=a\ast x_{i-1}+b$ 的$a$，$b$，$x0$。每次询问一个$v$，求出最小的$n$使得$x_n=v$

---

题解

首先简单推一下公式可得$\frac{x_n+\frac{b}{a-1}}{x_0+\frac{b}{a-1}}=a^n$
特判掉一些情况，剩下的就是BSGS模板题了。

BSGS公式：$a^x\equiv b(modp)$，设$x=kn+i$，则原方程变成$(a^n{)}^k=b(a^{-1}{)}^i$。
对于本题，每次询问只有$b$会发生改变，所以可以先预处理出来左边，复杂度会小很多。
一般我们$n$会取$\sqrt{p}$，那么复杂度为$O(\sqrt{p})$。但在这题就会TLE，所以我们对于$i$的枚举上限应该设为1e3左右，这样会让预处理的时候对$k$的枚举变大，但它只需要处理一次就行了。

然后我用了哈希，虽然unorderdmap也能用，但要跑3秒，哈希只要跑1秒。

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
const ll lim1=1e3+7;
const ll lim2=1e6+7;
ll x0;
ll n;
struct Hash {
    static const ll mod=1999997;
    static const ll N=1e6;
    ll head[mod+10],nx[N],top;
    ll hs[N],id[N];
    void init() {
        memset(head,-1,sizeof head);
        top=0;
    }
    void ist(ll x,ll y){
        ll k=x%mod;
        hs[top]=x;id[top]=y;nx[top]=head[k];head[k]=top++;
    }
    ll get(ll x){
        ll k=x%mod;
        for(ll i=head[k];i!=-1;i=nx[i]){
            if(hs[i]==x) return id[i];
        }
        return -1;
    }
}hs;
ll qpow(ll a,ll b,ll p){
    ll res=1;
    while(b){
        if(b&1) res=res*a%p;
        b>>=1;a=a*a%p;
    }
    return res;
}
void bsgs(ll a,ll p){
    hs.init();
    ll step=qpow(a,lim1,p);
    ll now=1;
    for(ll i=0;i<p;i+=lim1){
        if(hs.get(now)==-1) hs.ist(now,i);
        now=now*step%p;
    }
}
ll solve(ll inva,ll b,ll p){
    ll now=1;
    ll ret=n+1;
    for(ll i=0;i<=lim1;i++){
        ll tmp=b*now%p;
        if(hs.get(tmp)!=-1) ret=min(ret,hs.get(tmp)+i);
        now=now*inva%p;
    }
    return ret;
}

int main(){
    ll T,q;
    scanf("%lld",&T);
    ll a,b,p,v;
    while(T--){
        scanf("%lld%lld%lld%lld%lld%lld",&n,&x0,&a,&b,&p,&q);
        if(p==2){
            while(q--){
                scanf("%lld",&v);
                if(v==x0) printf("%0\n");
                else if(v==(x0*a+b)%p) printf("1\n");
                else printf("-1\n");
            }
            continue;
        }
        if(a==0){
            while(q--){
                scanf("%lld",&v);
                if(v==x0) printf("0\n");
                else if(v==b) printf("1\n");
                else printf("-1\n");
            }
            continue;
        }
        if(a==1){
            b=qpow(b,p-2,p);
            while(q--){
                scanf("%lld",&v);
                ll ans=(v-x0+p)%p*b%p;
                if(ans>=n) ans=-1;
                printf("%lld\n",ans);
            }
            continue;
        }
        bsgs(a,p);

        ll tmp=b*qpow(a-1,p-2,p)%p;
        ll xx=(x0+tmp)%p;
        xx=qpow(xx,p-2,p);

        ll inva=qpow(a,p-2,p);
        while(q--){
            scanf("%lld",&v);
            if(x0==v){printf("0\n");continue;}
            v=(v+tmp)%p*xx%p;
            ll ans=solve(inva,v,p);
            if(ans>=n) ans=-1;
            printf("%lld\n",ans);
        }
    }
}