离散对数随笔

阶

我们都知道欧拉定理，对于 $a\in \mathbb{Z}$，$m\in {\mathbb{N}}^{\ast }$，且 $(a,m)=1$，则有 $a^{\phi (m)}\equiv 1(\mathrm{mod}m)$，其中 $(a,m)$ 为 $\gcd (a,m)$ 的简写。即每经过 $\phi (m)$ 就绕了一圈。

但事实上可能存在更小的 $n|\phi (m)$，使得 $a^n\equiv 1(\mathrm{mod}m)$ 成立，最小的使得这个式子成立的 $n$ 称为 $a$ 模 $m$ 的阶，记作 ${\text{ord}}_m(a)$ 或者 ${\delta }_m(a)$。

它有一些性质：

• ${\delta }_m(a)\mid \phi (m)$
• ${\delta }_m(ab)={\delta }_m(a){\delta }_m(b)\leftrightarrow ({\delta }_m(a),{\delta }_m(b))=1$
• ${\delta }_m(a^k)=\frac{{\delta }_m(a)}{({\delta }_m(a),k)}$

这里忽略证明，感兴趣的话可以看 这里。

怎么计算 ${\delta }_m(a)$? 由于 $a^{\phi (m)}\equiv 1(\mathrm{mod}m)$，${\delta }_m(a)|\phi (m)$，所以可以记 $t=\phi (m)$，然后每次尝试把 $t$ 除以 $\phi (m)$ 的质因数，配合 Pollard-Rho 分解质因数是 $O(m^{\frac{1}{4}}+\log m)$ 的。

原根

我去，原

对于 $m\in \mathbb{Z}$，若有 $(g,m)=1$，满足 ${\delta }_m(g)=\phi (m)$，则称 $g$ 为 $m$ 的原根。

它也有一些性质：

• 一个正整数 $m$ 存在原根，当且仅当它可以表示成 $2,4,p^k,2p^k$ 中的一个。
• 若 $g$ 为 $m$ 的原根，则有 $g^{\frac{\phi (m)}{p}}\not\equiv 1(\mathrm{mod}m)$，其中 $p$ 为 $m$ 任意一个质因数。
• $m$ 最小原根的大小为 $O(m^{0.25})$。
• 若 $g$ 为 $m$ 的原根，且对于 $k\in {\mathbb{N}}^{\ast }$，有 $(k,\phi (m))=1$ ，则 $g^k$ 也是 $m$ 的一个原根。
• $m$ 若存在原根，则它的原根个数是 $O(\phi (\phi (m)))$。

第一点让我们知道一个数有没有原根（$O(m\log m)$）预处理，$O(1)$ 判断）。

第二点就是判定一个数是不是 $m$ 原根的方法（依次选取 $p$ 并判断）复杂度 $O(\log m)$。

第三点保证我们可以在 $O(m^{0.25}\log m)$ 时间内找到最小原根（暴力枚举）。

第四点和第五点让我们可以在 $O(\phi (m)\log m)$ 时间内找到其他原根。

于是就可以在 $O(m\log m)$ 时间内找出 $m$ 的所有原根。

板题：P6091 【模板】原根

离散对数

对于一个正整数 $m$ 和其原根 $g$（假设其存在），满足 $g^x\equiv b(\mathrm{mod}m)$，其中 $0\le x<\phi (m)$，则称 $x$ 为 $b$ 以 $g$ 为底模 $m$ 的指数，记作 ${\text{ind}}_{g}b$。显然这样的 $x$ 是唯一的。

它同样有一些性质：

对于 $(a,m)=1$，$(b,m)=1$，有

• ${\text{ind}}_{g}ab\equiv {\text{ind}}_{g}a+{\text{ind}}_{g}b(\mathrm{mod}\phi (m))$
• ${\text{ind}}_g{a}^k=k{\text{ind}}_{g}a(\mathrm{mod}\phi (m))$
• 若 $r$ 也为 $m$ 的原根，则 ${\text{ind}}_{g}a={\text{ind}}_{r}a\times {\text{ind}}_{g}r(\mathrm{mod}\phi (m))$
• 若 $a\equiv b(\mathrm{mod}m)$，则 ${\text{ind}}_{g}a={\text{ind}}_{g}b$

那么如何求解离散对数呢？

BSGS 算法

BSGS 全称 Baby-step Gaint-step，中文大步小步算法。对于正整数 $a$，$b$，$m$，它可以用来求解
$$a^x\equiv b(\mathrm{mod}m)$$
其中 $(a,m)=1$，$m$ 不一定为质数。

怎么做呢？令 $x=At-B$，其中 $1\le A\le \lceil \frac{\phi (m)}{t}\rceil$，$1\le B\le t$，这样 $At-B$ 能取遍 $0$ 到 $\phi (m)-1$。

于是有
$$a^{At}\equiv a^{B}b(\mathrm{mod}m)$$

于是可以枚举所有 $B$，把 $a^{B}b\mathrm{mod}m$ 可能的取值记下来（用哈希表或平衡树），再从小到大枚举 $A$，判断是否有和 $a^{At}$ 同余的 $a^{B}b$，若有，则答案为 $At-B$。如果全部枚举完都没有符合条件的，就说明无解。

显然 $t$ 取 $\lfloor \sqrt{m}\rfloor +1$ 时间复杂度最小。

代码：

map<ll, int> mp;
ll BSGS(ll a, ll b, ll p) {
    mp.clear(), a %= p, b %= p;
    ll cur = 1, t = sqrt(p) + 1;
    lp(i, 1, t) (cur *= a) %= p, mp[b * cur % p] = i;	// 枚举 A
    ll now = cur;
    lp(i, 1, t) {	// 枚举 B
        if(mp.count(now)) return i * t - mp[now];	// 找到了最小的答案
        (now *= cur) %= p;
    }
    return -INF;	// 无解
}

exBSGS

再看这个式子：
$$a^x\equiv b(\mathrm{mod}m)$$
若 $(a,m)=1$ 不一定成立，就需要用到 exBSGS。

注意到这个式子可以写成
$$a{a}^{x-1}+mk=b$$
由斐蜀定理，其有解的充要条件是 $b\mid (a,m)$，否则无解直接返回，两边同除以 $d=(a,m)$，得
$$\frac{a}{d}{a}^{x-1}+\frac{m}{d}k=\frac{b}{d}$$
即
$$\frac{a}{d}{a}^{x-1}\equiv \frac{b}{d}(\mathrm{mod}\frac{m}{d})$$
此时若 $(a,\frac{m}{d})=1$，就能用 BSGS 求解，只不过多了个 $\frac{a}{d}$ 系数，稍微改下 BSGS 就行。否则继续递归下去，进行 $k$ 次后变成
$$\frac{a^k}{D}{a}^{x-k}\equiv \frac{b}{D}(\mathrm{mod}\frac{m}{D})$$
其中 $D={\prod }_{i=1}^k{d}_i$。

那么这样的过程最多进行 max ⁡ { max ⁡ ( c i , c i ′ ) min ⁡ ( c i , c i ′ ) } = O ( log ⁡ m ) \max\{\frac{\max(c_i,c_i')}{\min(c_i,c_i')}\}=O(\log m) max{min(ci​,ci′​)max(ci​,ci′​)​}=O(logm) 次，其中 $c_i$ 为 $a$ 的质因数 $p_i$ 的幂次，$c_i^{\prime }$ 为 $m$ 的质因数 $p_i$ 的幂次。于是总时间复杂度为 $O(\sqrt{m}\text{polylog})$，若使用哈希就只有一个 log，使用 map 有两个 log。

需要注意若 $a^k$ 模 $m$ 为 $b$（没有除以 $d$ 的），就直接返回 $k$。

实现上，使用非递归实现，速度较快。

代码：

map<ll, int> mp;
ll BSGS(ll a, ll b, ll p, ll k = 1) {
    mp.clear(), a %= p, b %= p;
    ll cur = 1, t = sqrt(p) + 1;
    lp(i, 1, t) (cur *= a) %= p, mp[b * cur % p] = i;
    ll now = cur * k % p;	// 多乘了个系数
    lp(i, 1, t) {
        if(mp.count(now)) return i * t - mp[now];
        (now *= cur) %= p;
    }
    return -INF;	// 不能返回 -1，因为还加了 i + 1
}

ll exBSGS(ll a, ll b, ll m, ll k = 1) {
    ll A = a %= m, B = b %= m, M = m, cur = 1;
    for(int i = 0; ; ++i) {
        if(cur == B) return i;		// a ^ k % m = b 的情况
        (cur *= A) %= M; ll d = __gcd(a, m);
        if(b % d) return -INF;		// 判无解
        if(d == 1) return BSGS(a, b, m, k * a % m) + i + 1;
        b /= d, m /= d, k = k * a / d % m;
    }
}

板题：【模板】扩展 BSGS/exBSGS