BSGS算法 -＞ 高次同余方程问题

【算法分析】
BSGS（baby-step giant-step），即大步小步算法（谐称北上广深算法、拔山盖世算法）。可以较高效的求解形如a^x≡b(mod p)（p是素数）的高次同余方程。其中，符号“≡”表示同余运算。所谓同余运算，即如果两个整数a和b之差能被p整除，那么就说a和b对p同余，记作a≡b(mod p)。例如，100和60的差能被8整除，记作100≡60(mod 8)。
 

高次同余方程问题：已知整数a，b，p，其中 p 为质数且a、p互质，求最小的非负整数 x，满足 a^x≡b(mod p)。
思路：因为 p 为质数，根据费马小定理，a^(p−1)≡1(mod p)。又知道 a^0≡1(mod p)，所以幂有长度不超过 p 的循环节。暴力枚举小于 p 的数 x 即可。时间复杂度是O(p)。

当p较小时，上述算法可行。但是当 p 比较大时，就会TLE了。怎么办呢？这就需要利用BSGS算法求解高次同余方程了。BSGS算法的思想是 meet in the middle，时间复杂度O(sqrt(p))。

 

BSGS算法求解高次同余方程 a^x≡b(mod p)步骤如下：

1.令x=im−j，m=ceil(sqrt(p))，则有a^(im−j)≡b(mod p)

2.移项，得(a^m)^i≡b*a^j(mod p)

3.从 [0，m) 枚举j，将得到的b*a^j的值存入hash表；

4.从 [1，m] 枚举i，计算(a^m)^i。查第3步得到的hash表，如果有值与 i 相等，则得到的 im−j 就是x的最小值。

 

BSGS算法只能处理p是素数的情况，太特殊了，应用范围不大。而扩展BSGS算法可以解决p不是素数的一般情况。


【相关知识】
费马小定理：
若a为一个整数，p为一个质数，且a、p互质，则有a^(p-1)≡1(mod p)   
若a为一个整数，p为一个质数，且a是p的倍数，则有a^p≡a(mod p)     

最大公约数/最小公倍数 - https://blog.csdn.net/hnjzsyjyj/article/details/110733164
快速幂：https://blog.csdn.net/hnjzsyjyj/article/details/109556276


【算法实例】
洛谷P3846 - https://www.luogu.com.cn/problem/P3846
洛谷P2485 - https://www.luogu.com.cn/problem/P2485
洛谷P4454 - https://www.luogu.com.cn/problem/P4454
洛谷P4884 - https://www.luogu.com.cn/problem/P4884
POJ2417 - http://poj.org/problem?id=2417
POJ3243 - http://poj.org/problem?id=3243
BZOJ3122、HDU2815（利用扩展BSGS算法）


【算法代码】
// For 洛谷P3846 - https://www.luogu.com.cn/problem/P3846

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;

int fastPow(LL a,LL n,LL p) {
	LL ans=1;
	while(n) {
		if(n&1) ans=ans*a%p;
		a=a*a%p;
		n>>=1;
	}
	return ans;
}

LL BSGS(LL a,LL b,LL p) {
	map<LL,LL> hs;

	LL cur=b*a%p;
	LL m=ceil(sqrt(p));  //LL m=sqrt(p)+1;
	for(int j=1; j<=m; j++) {
		hs[cur]=j;
		cur=cur*a%p;
	}

	LL at=fastPow(a,m,p);
	LL now=at;
	for(int i=1; i<=m; i++) {
		if(hs[now])
			return i*m-hs[now];
		now=now*at%p;
	}
	return -1;
}

int main() {
	LL p,b,n;
	cin>>p>>b>>n;
	LL ans=BSGS(b,n,p);
	if(ans==-1)
		cout<<"no solution"<<endl;
	else
		cout<<ans<<endl;
	return 0;
}

/*
in:
5 2 3

out:
3
*/

【参考文献】
https://blog.csdn.net/clove_unique/article/details/50740412
https://www.bilibili.com/video/av287884055
https://oi-wiki.org/math/bsgs/
https://blog.csdn.net/zzkksunboy/article/details/73162229
https://www.cnblogs.com/pplppl/p/13392259.html
https://www.luogu.com.cn/blog/An-drew/post-bsgs
http://lbn187.is-programmer.com/posts/103186.html
https://www.acwing.com/blog/content/4597/
https://blog.csdn.net/haut_ykc/article/details/109274720