【算法】BSGS

$BSGS$ 又称大步小步算法拔山盖世，是一种求离散对数的算法。

---

$$a^x\equiv b(\mathrm{mod}p)$$

有这样的一个式子，且 $a$ 与 $p$ 互质，让你求 $x$ 的最小非负整数解为多少。

我们不妨设 $x$ 为 $i\sqrt{p}-j$，且 $i,j\in \left[0,\sqrt{p}\right]$，带入原式，可得：

$$a^{i\sqrt{p}-j}\equiv b(\mathrm{mod}p)$$

$$a^{i\sqrt{p}}\equiv b\times a^j(\mathrm{mod}p)$$

我们可以先枚举 $j$，将等式右边存入一个哈希表当中。再枚举 $i$，如果等式左边再模 $p$，意义下相等，则直接返回 $x$ 的值；如果一直找不到，那就说明无解。

#include<bits/stdc++.h>
#define int long long
using namespace std;
map<int,int>mp;
int a,b,p;
int Pow(int n,int k)
{
	int res=1;
	while(k)
	{
		if(k&1)
			(res*=n)%=p;
		(n*=n)%=p;
		k>>=1;
	}
	return res%p;
}
int BSGS(int a,int b,int p)
{
	int t=ceil(sqrt(p));
	for(int i=1;i<=t;i++)
		mp[(b*Pow(a,i))%p]=i;
	int k=Pow(a,t);
	for(int i=1;i<=t;i++)
	{
		int val=Pow(k,i);
		if(mp[val])
			return i*t-mp[val];
	}
	return -1;
}
signed main()
{
	scanf("%lld %lld %lld",&p,&a,&b);
	int ans=BSGS(a,b,p);
	if(ans==-1)
		puts("no solution");
	else printf("%lld",(ans%p+p)%p);
	return 0;
}