2019牛客多校第5场 generator 2（BSGS）

2019牛客多校第5场 generator 2（BSGS）

题目大意

已知$x_0$,a,b,p,现有递推式$x_n=(x_{n-1}\ast a+b)\%p$,每次询问一个v，问最小的n使得有$x_n=v$。

解题思路

我们首先可以发现这个数列可以写出通项
$$x_n=x_0\cdot a^n+b\frac{a^n-1}{a-1}（modp)$$
对于一个给出的v求解也就可以相应地写成
$$a^n\equiv \frac{(a-1)v+b}{(a-1)x_0+b}(modp)$$
对于这个问题除开a=0和a=1地特殊情况需要特殊讨论外，接下来用BSGS求解即可

需要注意的是这里对于每次给出的数列有上限1000次的询问，因此如果像常规做法时一样，大步小步都选择$\sqrt{(}p)$则最终会超时，因此本题中我们设定大步为$p^{\frac{2}{3}}$，对于每个给出的数列先预处理出大步，则每次询问都可以用小步的复杂度$p^{\frac{1}{3}}$解决

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
int quick_pow(int a,int b,int mod)
{
	int ans=1;
	while(b)
	{
		if(b&1) ans=1LL*ans*a%mod;
		a=1LL*a*a%mod;
		b>>=1;
	}
	return ans;
}
LL n;
int x0,a,b,p;
LL v;
#define int long long
unordered_map<LL,LL> mp;
int loop,up;
void pre_BSGS(int p,int b)
{
	mp.clear();
	up=ceil(pow(p,2.0/3));
	int t=1;
	for(int i=0;i<=up;i++) 
	{
		if(i==up) loop=t;
		mp[t]=i;
		t=1LL*t*b%p;
	}
}
int BSGS(int B,int N,int P)
{
	int m=ceil(pow(p,1.0/3));
	int obj=quick_pow(N,P-2,P);
	for(int i=1;i<=m;i++)
	{
		obj=1LL*obj*loop%P;
		if(mp.count(obj))
		{
			return 1LL*i*up-mp[obj];
		}
	}
	return -1;
}
#undef int
int main()
{
	int t;
	int q;
//	freopen("c.in","r",stdin);
	scanf("%d",&t);
	while(t--)
	{
		scanf("%lld%d%d%d%d",&n,&x0,&a,&b,&p);
		LL ab_apo=1LL*b*quick_pow(a-1,p-2,p)%p;
		LL x0ab_apo=(x0+ab_apo)%p;
		LL o_b=quick_pow(b,p-2,p);
		pre_BSGS(p,a);
		scanf("%d",&q);
		while(q--)
		{
			scanf("%lld",&v);
			if(a==1)
			{
				LL ans=(1LL*(v-x0)*o_b%p+p)%p;
				printf("%lld\n",ans<n?ans:-1);
				continue;
			}
			if(a==0)
			{
				if(x0==v) puts("0");
				else if(b==v) puts("1");
				else puts("-1");
				continue;
			}
			v=(v+ab_apo)%p;
			v=1LL*v*quick_pow(x0ab_apo,p-2,p)%p;
			LL ans=BSGS(a,v,p);
			printf("%d\n",ans<n?ans:-1);
		}
	}
}