「Luogu3306」[SDOI2013]随机数生成器

「Luogu3306」[SDOI2013]随机数生成器

problem

Solution

我们来回忆一下数列递推公式求通项的方法

令\(Y_i=X_i+\frac{b}{a-1}\)，则有

\[Y_i=aY_{i-1}\Rightarrow Y_i=a^{i-1}Y_1 \]

于是题目转化为求解方程

\[t+\frac b{a-1}\equiv a^{x-1}(X_1+\frac b{a-1})\pmod p \]

\[\Rightarrow a^{x-1}\equiv\frac{t+\frac b{a-1}}{X_1+\frac b{a-1}}\pmod p \]

好的，这个东西我们只要用BSGS就可以了

End?

---

几个细节：

\(X_1=t\)时，直接输出\(1\)

当\(a=0\)时，输出若\(b=t\)则输出\(2\)，否则无解，\(X_1=t\)的情况已经特判

当\(a=1\)时，方程实际上为

\[(x-1)b\equiv t-X_1\pmod p \]

根据裴蜀定理判掉无解，然后直接算\(x-1\)即可

需要注意的是如果\(\frac{t-X_1} b=p\)，则不需要取模直接输出

True End

Code

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <iostream>
#include <map>
#define mp(x,y) (make_pair((x),(y)))
#define inv(x) (fastpow((x),p-2))
using namespace std;
typedef long long ll;

template <typename T>void read(T &t)
{
	t=0;int f=0;char c=getchar();
	while(!isdigit(c)){f|=c=='-';c=getchar();}
	while(isdigit(c)){t=t*10+c-'0';c=getchar();}
	if(f)t=-t;
}

int T;

ll p,a,b,X1,t;
ll m;

ll fastpow(ll a,ll b)
{
	ll re=1,base=a;
	while(b)
	{
		if(b&1)
			re=re*base%p;
		base=base*base%p;
		b>>=1;
	}
	return re;
}

map<ll,ll> rec;
void Pre()
{
	rec.clear();
	ll tmp=1;
	for(register int i=0;i<m;++i,tmp=tmp*a%p)
		rec.insert(mp(tmp,i));
}

ll BSGS()
{
	ll iatm=inv(fastpow(a,m)),tmp=1;
	for(register int i=0;i<m;++i,tmp=tmp*iatm%p)
		if(rec.find(b*tmp%p)!=rec.end())
			return i*m+rec[b*tmp%p];
	return -2;
}

ll gcd(ll a,ll b)
{
	return b?gcd(b,a%b):a;
}

int main()
{
	read(T);
	while(T--)
	{
		read(p),read(a),read(b),read(X1),read(t);
		if(X1==t)
		{
			puts("1");
			continue;
		}
		if(a==1)
		{
			t=(t-X1+p)%p;
			if(t%gcd(b,p))
				puts("-1");
			else
				printf("%lld\n",t*inv(b)+1==p?p:(t*inv(b)+1)%p);
			continue;
		}
		if(a==0)
		{
			printf("%lld\n",b==t?2:-1ll);
			continue;
		}
		b=b*inv(a-1)%p;
		b=(t+b)*inv(X1+b)%p;
		m=ceil(sqrt(p));
		Pre();
		printf("%lld\n",BSGS()+1);
	}
	return 0;
}