BSGS

BSGS

问题：给定整数 $a,b,p$，其中 $a,p$ 互质，求方程 $a^x\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$ 的解。

由于 $a,p$ 互质，所以我们可以在模 $p$ 意义下执行关于 $a$ 的乘除运算。

我们将 $x$ 表示为 $i\times t-j$（$t=\lceil \sqrt{p}\rceil$，$i\in [0,t]$，$j\in [0,t-1]$），原方程就变成 $a^{i\times t-j}\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$，也即 $(a^t{)}^i\equiv b\times a^j(\mathrm{m}\mathrm{o}\mathrm{d}p)$。

这样表示的话，显然 $x\in [0,p-1]$，至于为什么，在下文中有解释，先接着看吧。

然后我们枚举 $j\in [0,t-1]$，将得到的 $b\times a^j\mathrm{m}\mathrm{o}\mathrm{d}p$ 插入一个哈希表中（也可以用 unordered_map 维护）。

然后我们枚举 $i\in [0,t]$，计算出 $(a^t{)}^i\mathrm{m}\mathrm{o}\mathrm{d}p$，在哈希表中查找是否有对应的 $j$，如果有，那么这时的 $i\times t-j$ 就是答案，直接返回就行，否则就继续枚举。

时间复杂度为 $O(\sqrt{n})$。

现在我们来解决一下上文留下的问题。
先来证明一个公式：$a^{k\mathrm{m}\mathrm{o}\mathrm{d}(p-1)}\equiv a^k(\mathrm{m}\mathrm{o}\mathrm{d}p)$。
$k\mathrm{m}\mathrm{o}\mathrm{d}(p-1)$ 也就是 $k-m(p-1)$，原式可以变成 $a^{k-m(p-1)}\equiv a^k(\mathrm{m}\mathrm{o}\mathrm{d}p)$。
继续化简，变成 $\frac{a^k}{a^{m(p-1)}}\equiv a^k(\mathrm{m}\mathrm{o}\mathrm{d}p)$。
也就是说，如果 $a^{m(p-1)}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$，那么式子就成立。
由费马小定理知，当 $p$ 为质数且 $\gcd (a,p)=1$ 时，$a^{p-1}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$。
也即，$a^{m(p-1)}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$，故原式得证。
所以说，如果原方程有解的话，那在 $[0,p-1]$ 中必然有一个解，所以枚举到 $p$ 就可以了。

模板题：[TJOI 2007] 可爱的质数

exBSGS

很多时候，$a,p$ 是不互质的，这时 $\mathrm{B}\mathrm{S}\mathrm{G}\mathrm{S}$ 就不再适用，我们就需要把这个算法扩展一下。

设 $d=\gcd (a,p)$，显然 $d>1$，令 $a=a^{\prime }\times d$，$p=p^{\prime }\times d$，则：

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

不难发现，如果 $b$ 不是 $d$ 的倍数，那么只有当 $b=1$ 时才有解（此时 $x=0$），否则就无解。因此可以把 $b=1$ 的情况特判一下，其他情况下只有 $b$ 是 $d$ 的倍数时才有解，那就令 $b=b^{\prime }\times d$，则：

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

消去一个 $d$，则：

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

把 $a^{\prime }$ 又换成 $\frac{a}{d}$，得到

$$\frac{a}{d}\times a^{x-1}\equiv b^{\prime }(\mathrm{m}\mathrm{o}\mathrm{d}{p}^{\prime })$$

由于这样处理之后，只能保证 $a^{\prime }$ 和 $p^{\prime }$ 是互质的，并不能说 $a$（即 $a^{\prime }\times d$） 和 $p^{\prime }$ 是互质的，所以要继续这样处理，直到 $\gcd (a,p)=1$。假如我们做了 $num$ 次处理，且每次的 $d$ 之积为 $\mathrm{D}$，则：

$$\frac{a^{num}}{\mathrm{D}}\times a^{x-num}\equiv \frac{b}{\mathrm{D}}(\mathrm{m}\mathrm{o}\mathrm{d}\frac{p}{\mathrm{D}})$$

此时的 $\gcd (a,\frac{p}{\mathrm{D}})=1$，就可以用 $\mathrm{B}\mathrm{S}\mathrm{G}\mathrm{S}$ 了。

注意我们这样解出来是 $x\ge num$ 的解，还要特判一下 $[0,num-1]$ 中是否有合法解。

而且我们解出来的实际上是 $x-num$ 的值，因此最后还要加上一个 $num$。

模板题：[SPOJ 3105] Mod