generator 2

题目链接
牛客多校 第五场
题目大意就是给出 x0,a,b,p,n, 求出一个数v在x的数列中最先出现的下标是多少
其中 $x_i=a\ast x_{i-1}+b$
通过题目中给的式子 我们可以化简得到：
$x_n=x_0\ast a^n+b\ast \frac{(a^{n-1}-1)}{(a-1)}$
将v带入化简我们可以得到
$a^n=\frac{v\ast (a-1)+1}{x0\ast (a-1)+b}$
根据BSGS 公式为 $a^n\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}m)$
我们可将其化简为$(a^t{)}^i\equiv b\ast a^j(\mathrm{m}\mathrm{o}\mathrm{d}m)$
具体得证明 可以去网上查一查,将其带入本题那么 式子就可以化简为
$(a^t{)}^i=a^j\ast \frac{v\ast (a-1)+1}{x0\ast (a-1)+b}⟺(a^t{)}^i\ast \frac{x0\ast (a-1)+b}{v\ast (a-1)+1}=a^j$
带入我们枚举将其存起来,然后在枚举i一一查询找到返回即可,注意我们不能向平常一样t取$\sqrt{p}$
这里应该取$p\ast \frac{2}{3}$ 即j的上界，i从0到$p\ast \frac{1}{3}$
代码：

#include <bits/stdc++.h>
using namespace std;
#define int long long
int n,x0,a,b,p;
unordered_map<int ,int> mp;
int power(int a, int b,int mod){
    int ans = 1;
    while(b){
        if(b&1){
            ans = a*ans %mod;
        }
        b >>= 1;
        a = a*a%mod;
    }
    return ans%mod;
}
int flag;
int t,t1;
void pre_BSGS(int a, int p){
    mp.clear();
    a %= p;
    t = pow(p, 1.0/3.0) + 1;
    t1 = pow(p, 2.0/3.0) + 1;
    int val = 1;
    for(int j = 0; j <= t1; j ++){
        if(j == t1) flag = val;
        mp[val] = j;
        val = val*a%p;
    }
}

int BSGS(int a, int b, int p){
    for(int i = 1; i <= t; i ++){
        b = b*flag%p;
        if(mp.count(b))
        {
            return i*t1 - mp[b];
        }
    }
    return -1;

}
signed main()
{
    int t;
    scanf("%lld",&t);
    while(t --){
        scanf("%lld%lld%lld%lld%lld",&n,&x0,&a,&b,&p);
        pre_BSGS(a, p);
        int q;
        scanf("%lld",&q);
        while(q --){
            int v;
            scanf("%lld",&v);
            if(a == 1){
                if(v == x0){
                    printf("0\n");
                }
                else{
                    int temp = (v - x0 + p) % p;
                    if(temp/b < n){
                        printf("%lld\n",temp/b);
                    }
                    else puts("-1");
                }
            }
            else if(a == 0){
                if(v == x0){
                    printf("0\n");
                }
                else if(b == v&&n >= 1){
                    printf("1\n");
                }
                else printf("-1\n");
            }
            else{
                int temp = power(((v%p*(a - 1 + p)%p)%p+ b)%p,p-2,p) *((x0%p*(a - 1 + p)%p)%p + b)%p;
                int ans =  BSGS(a, temp, p);
                if(ans < n) printf("%lld\n",ans);
                else printf("-1\n");
            }
        }

    }
}
/*

45
1000000000000000000 1 5 0 1000000007
5
1
10
1000000006
12345678
1234567*/