【BZOJ3122】[Sdoi2013]随机数生成器 BSGS+exgcd+特判

【BZOJ3122】[Sdoi2013]随机数生成器

Description

Input

输入含有多组数据,第一行一个正整数T,表示这个测试点内的数据组数。  
 
接下来T行,每行有五个整数p,a,b,X1,t,表示一组数据。保证X1和t都是合法的页码。 

注意:P一定为质数

Output

共T行,每行一个整数表示他最早读到第t页是哪一天。如果他永远不会读到第t页,输出-1。 

Sample Input

3
7 1 1 3 3
7 2 2 2 0
7 2 2 2 1

Sample Output


1
3
-1

HINT

0<=a<=P-1,0<=b<=P-1,2<=P<=10^9

题解:又一道特判神题~

若A=0,直接判;若A=1,用exgcd求;若A>1,此时要用到高中数学的知识。

此时xi+c变成了等比数列

然后上BSGS就行了

#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <map>
using namespace std;
typedef long long ll;
map<ll,ll> mp;
ll pm(ll x,ll y,ll z)
{
	ll ret=1;
	while(y)
	{
		if(y&1)	ret=ret*x%z;
		x=x*x%z,y>>=1;
	}
	return ret;
}
ll solve(ll A,ll B,ll P)
{
	if(!A&&!B)	return 0;
	if(!A)	return -1;
	ll i,x,y,m;
	mp.clear(),mp[B]=0,m=ceil(sqrt(P));
	for(x=1,i=1;i<=m;i++)	x=x*A%P,mp[x*B%P]=i;
	for(y=1,i=1;i<=m;i++)
	{
		y=y*x%P;
		if(mp.find(y)!=mp.end())	return i*m-mp[y];
	}
	return -1;
}
ll exgcd(ll a,ll b,ll &x,ll &y)
{
	if(b==0){x=1,y=0;	return a;}
	ll tmp=exgcd(b,a%b,x,y),t=x;
	x=y,y=t-a/b*x;
	return tmp;
}
int main()
{
	ll T,A,B,P,X,t,tmp,xx,yy;
	scanf("%lld",&T);
	while(T--)
	{
		scanf("%lld%lld%lld%lld%lld",&P,&A,&B,&X,&t);
		if(X==t)
		{
			printf("1\n");
			continue;
		}
		if(A==0)
		{
			if(B==t)	printf("2\n");
			else	printf("-1\n");
			continue;
		}
		if(A==1)
		{
			if(B==0)
			{
				printf("-1\n");
				continue;
			}
			ll g=exgcd(B,P,xx,yy),C=t+B-X;
			if(C%g)
			{
				printf("-1\n");
				continue;
			}
			C/=g,P/=g,xx=xx*C%P;
			if(xx<=0)	xx+=P;
			printf("%lld\n",xx);
			continue;
		}
		tmp=solve(A,((A-1)*t+B)%P*pm(((A-1)*X+B)%P,P-2,P)%P,P)+1;
		if(!tmp)	printf("-1\n");
		else	printf("%lld\n",tmp);
	}
	return 0;
}

| 欢迎来原网站坐坐! >原文链接<

posted @ 2017-06-13 10:26  CQzhangyu  阅读(217)  评论(0编辑  收藏  举报