BSGS、EXBSGS

大步小步算法英文名： baby-step gaint-step (BSGS) .

该算法可在$\Theta \left(\sqrt{p}\right)$的时间复杂度内求解
$a^x\equiv bmodp$，其中$a$与$p$互质，方程的解$x$满足$0\le x\le p$

令$x=A\lceil \sqrt{p}\rceil -B$，其中$0\le A,B\le \lceil \sqrt{p}\rceil$

则有$a^{A\lceil \sqrt{p}\rceil -B}\equiv b$，即$a^{A\lceil \sqrt{p}\rceil }\equiv b{a}^B$

由于a、b是确定的，所以我们可以先枚举右边$b{a}^B$存入hash/map中，再枚举左边$a^{A\lceil \sqrt{p}\rceil }$的A，从而计算出所有的x

P3846 [TJOI2007] 可爱的质数/【模板】BSGS

代码如下

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;

int main()
{
	int p, a, b;
	cin >> p >> a >> b;
	
	int m = (int)sqrt(p);
	if(m * m != p) m ++;
	
	map<int, int> hash;
	int s = 1;
	for(int i = 1; i <= m; ++ i)
	{
		s = (LL)s * a % p;
		b = (LL)b * a % p;
		hash[b] = i;
	}
	
	bool flag = false;
	a = s;
	for(int i = 1; i <= m; ++ i)
	{
		if(hash[a])
		{
			printf("%d\n", i * m - hash[a]);
			flag = true;
			break;
		}
		
		a = (LL)a * s % p;
	}
	
	if(!flag) printf("no solution");
	
	return 0;
}

---

扩展大步小步算法

上边的算法只针对$a$与$p$互质的情形，如果两者不互质，怎么计算？

当$x>0$时，$a^x\equiv b(modp)$等价于$a^{x-1}a+np=b$由裴蜀定理可知该方程有解的条件是$b$是$gcd(a,p)$的倍数

设$d=gcd(a,p)$，则有$\frac{a^{x-1}a}{b}+\frac{np}{d}=\frac{b}{d}$，

即$\frac{a}{b}{a}^{x-1}\equiv \frac{b}{d}(mod\frac{p}{d})$

这时如果$gcd(a,\frac{p}{d})=1$，就可以直接使用BSGS求解，左侧多出一个系数，修改一下BGSG即可，解出的$x$加上$1$即是原方程的一个解，否则可以继续递归下去，直到互质为止

BSGS算出来的是最小解，而扩展BSGS算出来的不一定是最小解

注意：递归时需要特判，如果$b\equiv 1$直接返回$x=0$即可，当$a\equiv b\equiv 0$时应该返回1，一般认为$0^0=1$

时间复杂度：$\Theta (\sqrt{n}logn)$

P4195 【模板】扩展BSGS

具体代码如下

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;

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

map<int, int> hs;
int BSGS(int a, int b, int p, int k)
{
	hs.clear();
	int m = sqrt(p) + 1, s = 1;
	for(int B = 1; B <= m; ++ B)
	{
		s = 1ll * s * a % p;
		hs[1ll * b * s % p] = B;
	}
	
	k = 1ll * k * s % p;
	for(int i = 1; i <= m; ++ i)
	{
		int j = hs.count(k) ? hs[k] : -1;
		if(j >= 0 && i * m - j >= 0) return i * m - j;
		k = 1ll * k * s % p;
	}
	return -1;
}

int exBSGS(int a, int b, int p)
{
	if(b == 1 || p == 1) return 0;
	int d = gcd(a, p), k = 1, cnt = 0;
	while(d != 1)
	{
		if(b % d) return -1;
		cnt ++;
		b /= d, p /= d;
		k = (LL)k * a / d % p;
		if(k == b) return cnt;
		d = gcd(a, p); 
	}
	
	int res = BSGS(a, b, p, k);
	if(res < 0) return -1;
	return res + cnt;
}

int main()
{
	int a, b, p;
	while(~scanf("%d%d%d", &a, &p, &b), (a || p || b))
	{
		int x = exBSGS(a % p, b % p, p);
		if(x < 0) puts("No Solution");
		else printf("%d\n", x);
	}
	
	return 0;
}

上边exBSGS函数中有一句if(k == b) return cnt,转化为公式是

$\frac{a^{cnt}}{d^{cnt}}\times a^{x-cnt}\equiv \frac{b}{d^{cnt}}(mod\frac{p}{d^{cnt}})$

其中$\frac{a^{cnt}}{d^{cnt}}$就是$k$，$\frac{b}{d^{cnt}}$就是$b$，$k=b$时相当于

$\frac{a^{cnt}}{d^{cnt}}\equiv \frac{b}{d^{cnt}}(mod\frac{p}{d^{cnt}})$

这个等同于

$a^{cnt}\equiv b(modp)$

此时cnt就是我们要求的$x$，直接返回即可