大步小步算法（BSGS）详解

最新推荐文章于 2022-11-15 21:18:43 发布

LZRcqbz

最新推荐文章于 2022-11-15 21:18:43 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

分类专栏： # 数论

本文链接：https://blog.csdn.net/qq_37656398/article/details/101228362

数论专栏收录该内容

8 篇文章

订阅专栏

本文深入讲解了BSGS算法，一种高效解决高次同余方程Ax≡B(mod P)的方法，其中A、B、P为已知数，P为质数。通过分析哈希表的应用和算法复杂度，介绍了如何在[0,P-1)范围内寻找最小解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

BSGS（Baby Step Gaint Step）算法详解

BSGS算法是用于求解形如 $P)A^x\equiv B(\mod P)$ 的高次同余方程，其中 $A, B, P$ 为已知，且要求 $gcd⁡(A,P)=1\gcd(A,P)=1$ 是一个质数。

前置知识

哈希

算法分析

先证明一个结论：若原方程有解，则 $x∈[0,P−1)x\in[0,P-1)$ 一定有解。

因为 $gcd⁡(A,P)=1\gcd(A,P)=1$ ，由费马小定理得 $P)A^{P-1}\equiv1(\mod P)$ 。

则有 $P)A^{x+k(P-1)}\equiv A^x(\mod P)$ ，其中 $k$ 为整数。

换句话说就是出现了循环。这样一来，这个方程的最小解就一定在 $[0, P - 1)$ 中了。

我们设 $x = a m - b$ ，其中 $m$ 为任意一个正数，且有 $a∈[0,m)a\in[0,m)$ 。

那么变一下形就发现 $P)A^{am}\equiv B\cdot A^{b}(\mod P)$ 。

我们暴力算出右边的 $PB\cdot A^b\mod P$ 的所有可能的值，并存起来。

然后再枚举左边的 $a$ ，并查询在右边有没有相等的值。

所以这个算法的复杂度为 $O(max⁡(m,Pm))O(\max(m,\frac{P}{m}))$ 。可以证明当 $m=Pm=\sqrt P$ 时，算法的复杂度达到下界，为 $O(P)O(\sqrt P)$ 。

~~就这么简单QAQ…~~

当然为了查询方便，我们写一个哈希或者直接用map存下右边所有的值即可。

这里建议用哈希，因为用map的话会多出一个 $log⁡\log$ 。

模版

struct HashTable {
	struct HashNode {
		int u, v;
		HashNode *nxt;
	};
	HashNode pool[HashMod * 20 + 5];
	HashNode *hcnt, *head[HashMod + 5];
	void clear() {
		hcnt = &pool[0], memset(head, 0, sizeof head);
	}
	void addnode(int u, int v, int w) {
		HashNode *p = ++hcnt;
		p->u = w, p->v = v;
		p->nxt = head[u], head[u] = p;
	}
	void hash(int x, int k) {
		int s = x % HashMod;
		addnode(s, k, x);
	}
	int query(int x) {
		int s = x % HashMod;
		for(HashNode *p = head[s]; p != NULL; p = p->nxt)
			if(p->u == x) return p->v;
		return -1;
	}
};
HashTable hashtab;
ll BabyStepGaintStep(ll y, ll z, ll p) {
	if(y % p == 0) return -1;
	y %= p, z %= p;
	if(z == 1) return 0;
	hashtab.clear();
	ll m = (ll) sqrt(p) + 0.5;
	for(ll i = 0, t = z; i < m; i++, t = t * y % p)
		hashtab.hash(t, i);
	for(ll i = 1, tmp = QuickPow(y, m, p), t = tmp; i <= m + 1; i++, t = t * tmp % p) {
		ll k = hashtab.query(t);
		if(k == -1) continue;
		return i * m - k;
	}
	return -1;
}