bsgs(求解高次同余方程)学习笔记

基础篇

BSGS（baby-step giant-step），即大步小步算法。常用于求解离散对数问题。形式化地说，该算法可以在 $O(\sqrt{p})$ 的时间内求解

$$a^x\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

其中 $a\perp p$。方程的解 $x$ 满足 $0\le x<p$。（在这里需要注意，只要 $a\perp p$ 就行了，不要求 $p$ 是素数）

例题：3846 [TJOI2007] 可爱的质数/【模板】BSGS
求解过程
要求$a^x\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$
首先令$x=im-j$
化为$(a^m{)}^i\equiv b\ast a^j(\mathrm{m}\mathrm{o}\mathrm{d}p)$
j的取值范围是$[0,m)$，先枚举等号右边，并存入hash表中
i的取值范围是$[1,m]$，枚举等号左边，同时在hash表中查找是否有对应的值，找到的第一个为最小

#include<bits/stdc++.h>
#include <unordered_map>
using namespace std;
template<class...Args>
void debug(Args... args) {//Parameter pack
	auto tmp = { (cout << args << ' ', 0)... };
	cout << "\n";
}
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
typedef pair<int, int>pii;
const ll N = 1e5 + 5;
const ll INF = 0x7fffffff;
const ll MOD = 1e9 + 7;
ll qpow(ll x, ll n, ll mod) {
	ll ans = 1;
	while (n) {
		if (n & 1)ans = (ans * x) % mod;
		x = (x * x) % mod;
		n >>= 1;
	}
	return ans % mod;
}
ll bsgs(ll a, ll b, ll p) {
	if (b == 1) return 0;
	ll m = ceil(sqrt(p));//时间复杂度为O(sqrt(n))
	ll t = b;
	unordered_map<ll, ll> dict;
	dict[b] = 0;
	for (int j = 1; j < m; ++j) {//计算b*aj对应的值将对应的j值存入hash表中
		t = t * a % p;
		dict[t] = j;
	}
	ll mi = qpow(a, m, p);//计算a^m
	t = 1;
	for (int i = 1; i <= m; ++i) {//枚举i
		t = t * mi % p;
		if (dict.count(t) != 0) return i * m - dict[t];
	}
	return -1;
}
int main() {
	ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);

	ll b, n, p;
	cin >> p >> b >> n;
	ll ans = bsgs(b, n, p);
	if (ans == -1)cout << "no solution\n";
	else cout << ans << "\n";

	return 0;
}

进阶篇

求解

$$x^a\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

其中 $p$ 是个质数。

该模型可以通过一系列的转化为成 基础篇 中的模型，你可能需要了解关于 阶与原根 的知识。

由于式子中的模数 $p$ 是一个质数，那么 $p$ 一定存在一个原根 $g$。因此对于模 $p$ 意义下的任意的数 $x(0\le x<p)$ 有且仅有一个数 $i(0\le i<p-1)$ 满足 $x=g^i$。

方法一

我们令 $x=g^c$，$g$ 是 $p$ 的原根（我们一定可以找到这个 $g$ 和 $c$），问题转化为求解 $(g^c{)}^a\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$。稍加变换，得到

$$(g^a{)}^c\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

于是就转换成了我们熟知的 BSGS 的基本模型了，可以在 $O(\sqrt{p})$ 解出 $c$，这样可以得到原方程的一个特解 $x_0\equiv g^c(\mathrm{m}\mathrm{o}\mathrm{d}p)$。

方法二

我们仍令 $x=g^c$，并且设 $b=g^t$，于是我们得到

$$g^{ac}\equiv g^t(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

方程两边同时取离散对数得到

$$ac\equiv t(\mathrm{m}\mathrm{o}\mathrm{d}\phi (p))$$

我们可以通过 BSGS 求解 $g^t\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$ 得到 $t$，于是这就转化成了一个线性同余方程的问题。这样也可以解出 $c$，求出 $x$ 的一个特解 $x_0\equiv g^c(\mathrm{m}\mathrm{o}\mathrm{d}p)$。

找到所有解

在知道 $x_0\equiv g^c(\mathrm{m}\mathrm{o}\mathrm{d}p)$ 的情况下，我们想得到原问题的所有解。首先我们知道 $g^{\phi (p)}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$，于是可以得到

$$\forall t\in \mathbb{Z},x^a\equiv g^{c\cdot a+t\cdot \phi (p)}\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

于是得到所有解为

$$\forall t\in \mathbb{Z},a\mid t\cdot \phi (p),x\equiv g^{c+\frac{t\cdot \phi (p)}{a}}(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

对于上面这个式子，显然有 $\frac{a}{\gcd (a,\phi (p))}\mid t$。因此我们设 $t=\frac{a}{\gcd (a,\phi (p))}\cdot i$，得到

$$\forall i\in \mathbb{Z},x\equiv g^{c+\frac{\phi (p)}{\gcd (a,\phi (p))}\cdot i}(\mathrm{m}\mathrm{o}\mathrm{d}p)$$

这就是原问题的所有解。