#Baby Step,Giant Step# [luogu P3846] [TJOI2007] 可爱的质数/【模板】BSGS

Title

https://www.luogu.com.cn/problem/P3846

**大意：**给定整数$a,b,p$，其中$a,p$互质，求一个非负整数$x$，使得$$a^x\equiv b(modp)$$

---

Solution

Baby Step,Giant Step大步小步算法
设$x=i\ast t-j$，其中$t=|\sqrt{p}|,0\le j\le t-1$，则方程变成$a^{i\ast t-j}\equiv b(modp)$
对于所有的$j\in [0,t-1]$，把$b\ast a^{j}modp$插入一个$Hash$表。
枚举$i$的所有可能取值，即$i\in [0,t]$,计算出$(a^t{)}^{i}modp$，在$Hash$表中查找是否存在对应的$j$，更新答案即可。时间复杂度为$O(\sqrt{(}p))$

---

Code

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<map>
#define ll long long 
using namespace std; 
inline ll ksm(ll a,ll b,ll p){
	ll ans=1; 
	for(;b;b>>=1,a=(a*a)%p) if (b&1) ans=(ans*a)%p; 
	return ans; 
}
inline ll baby_step_giant_step(ll a,ll b,ll p){
	map<ll,ll>hash; 
	hash.clear(); 
	b%=p; 
	ll t=(ll)sqrt(p)+1; 
	for(ll j=0;j<t;j++){
		ll val=(long long)b*ksm(a,j,p)%p; 
		hash[val]=j; 
	}
	a=ksm(a,t,p); 
	if (a==0) return b==0?1:-1; 
	for(ll i=0;i<=t;i++){
		ll val=ksm(a,i,p); 
		ll j=hash.find(val)==hash.end()?-1:hash[val]; 
		if (j>=0&&i*t-j>=0) return i*t-j; 
	}
	return -1; 
}
int main(){
	ll a,b,p; 
	scanf("%lld%lld%lld",&p,&a,&b); 
	ll ans=baby_step_giant_step(a,b,p); 
	if (ans!=-1) printf("%lld",ans); else printf("no solution"); 
}