数论:从提高组到提高组

说明

最近可爱的 MGJ 连上了七天的数论课,然后她写了一篇提高组数论的合集。

由于篇幅原因,大部分例题不放代码。

此文章部分参考他人文章或借他人文章进行优化,链接如下:

  1. https://www.luogu.com.cn/problem/solution/P5656
  2. https://www.luogu.com.cn/article/fcv6pwn2
  3. https://www.luogu.com.cn/article/n6if1g3y

更新日志:

  • 2026.3.4 22:35:更新扩展欧几里得(Exgcd);
  • 2026.3.5:更新乘法逆元;
  • 2026.3.7:更新欧拉函数;
  • 2026.3.8-3.9:更新中国剩余定理(CRT);
  • 2026.3.9-3.10:更新欧拉定理;
  • 2026.3.11:更新小步大步算法(BSGS);
  • 2026.3.11 21:36:完善所有内容,整改错别字。

本文耗时 \(7\) 天在每天课余时间抽空写完,作者写文章不易,如有不当之处敬请谅解(可以私信作者)。

扩展欧几里得(Exgcd)

Part 0:前置知识

  • 基础数学(四则运算,符号定义等);
  • 欧几里得定理:\(\gcd(a, b) = \gcd(b,\ a \bmod b)\);
  • 裴蜀定理。

Part 1:扩展欧几里得的定义

扩展欧几里得是一个算法,一般解决如下核心问题:

给定两个整数 \(a, b\),找到整数 \(x, y\) 满足:

\[ax + by = \gcd(a, b) \]

这个方程叫做裴蜀方程(Bézout's identity)。

它在信息学竞赛中有着广泛应用,如求解一次不定方程或同余方程。

Part 2:裴蜀方程的求解

我们使用递归推导。

我们对 \(ax + by = \gcd(a, b)\) 用一次欧几里得定理,得:

\[bx_1 + (a \bmod b)y_1 = \gcd(b,\ a \bmod b) \]

将 \(a \bmod b\) 转化为 \(a - b \cdot \left\lfloor \frac{a}{b} \right \rfloor\),得:

\[ax + by = bx_1 + \left(a - b \cdot \left\lfloor \frac{a}{b} \right \rfloor\right)y_1 = bx_1 + ay_1 - b \cdot \left\lfloor \frac{a}{b} \right \rfloor y_1 = ay_1 + b\left(x_1 - \left\lfloor \frac{a}{b} \right \rfloor y_1\right) \]

与原式比较,得到一组特解:

\[\begin{cases} x = y_1 \\ y = x_1 - \left\lfloor \frac{a}{b} \right \rfloor y_1 \end{cases} \]

于是,我们找到了上层 \(x, y\) 和本层 \(x_1, y_1\) 的关系。

最终当 \(b = 0\) 时,\(\gcd(a, 0) = a\),方程变为 \(ax + 0y = a\),显然得到一组解:\(x = 1\),\(y = 0\)。

注意:

信息学大忌:在 \(b = 0\) 时耍帅把 \(y\) 随便写(如 \(114514\)),因为每一层递归时 \(x, y\) 都会变化。当递归次数足够多时,\(x, y\) 将会超出变量范围或产生某些其他问题。

下面给出两种模板代码,第一种好理解但代码较长(也长不了多少),新手建议使用第一种:

Code 1:

cpp 复制代码
ll Exgcd(ll a, ll b, ll &x, ll &y) {
  if (!b) {
    x = 1, y = 0;
    return a;
  }
  ll d = Exgcd(b, a % b, x, y), tmp = x;
  x = y, y = tmp - (a / b) * y;
  return d;
}

Code 2:

cpp 复制代码
ll Exgcd(ll a, ll b, ll &x, ll &y) {
  if (!b) {
    x = 1, y = 0;
    return a;
  }
  ll d = Exgcd(b, a % b, y, x);
  y -= a / b * x;
  return d;
}

Part 3:扩展欧几里得的应用

3.1:解不定方程

常见问题:求不定方程 \(ax + by = c\) 的一组解。

首先,先使用裴蜀定理判断是否有解:由裴蜀定理得 \(\gcd(a, b) \mid (ax + by)\),故 \(\gcd(a, b) \nmid c\) 时不定方程无解。

然后,用 Exgcd 求出 \(ax + by = \gcd(a, b)\) 的一组整数特解 \(x_0, y_0\)。那么有:

\[ax_0 + by_0 = \gcd(a, b) \]

两边同乘 \(\cfrac{c}{\gcd(a, b)}\),得:

\[a\cfrac{cx_0}{\gcd(a, b)} + b\cfrac{cy_0}{\gcd(a, b)} = c \]

与原式比较,可以得到一组特解:

\[\begin{cases} x_1 = \cfrac{cx_0}{\gcd(a, b)} \\ y_1 = \cfrac{cy_0}{\gcd(a, b)} \end{cases} \]

有了特解,那么就可以推导通解了。

显然有:

\[\begin{cases} ax_0 + by_0 = c \\ ax + by = c \end{cases} \]

令 \(\Delta x = x - x_0\),\(\Delta y = y - y_0\)。

则:

\[a(x_0 + \Delta x) + b(y_0 + \Delta y) = c \]

展开,得:

\[ax_0 + a\Delta x + by_0 + b\Delta y = c \]

又 \(ax_0 + by_0 = c\),则:

\[a\Delta x + b\Delta y = 0 \]

通过各种方法,解得:

\[\begin{cases} \Delta x = k \cdot \cfrac{b}{\gcd(a, b)} \\ \Delta y = -k \cdot \cfrac{a}{\gcd(a, b)} \end{cases} \]

于是,我们得到通解形式:

\[\begin{cases} x = x_0 + k \cdot \cfrac{b}{\gcd(a, b)} \\ y = y_0 - k \cdot \cfrac{a}{\gcd(a, b)} \end{cases} \]

其中 \(k \in \mathbb{Z}\)。

我们还可以推导出,如果要求最小非负整数解,可以提前将系数 \(a, b\) 都除以 \(\gcd(a, b)\)。

3.2:解同余方程

常见问题:解形如 \(ax \equiv c \pmod b\) 的同余方程。

考虑将此方程尽可能写成 \(ax + by = c\) 的形式:

\[\because ax \equiv c \pmod b \therefore ax \bmod b = c \bmod b \because by \bmod b = 0\ \ (y \in \mathbb{N}) \therefore ax \bmod b + by \bmod b = c \bmod b \therefore ax + by \equiv c \pmod b \]

于是就变成了不定方程的形式。

Part 4:例题

P1082 [NOIP 2012 提高组] 同余方程 / P2613【模板】有理数取余

两题均为套模板题。

讲解几个需要注意的点:

  • 在 P1082 中,为了避免 \(x_0\) 为负数,我们需要先 \(x \to x \bmod b\),再将 \(x \to x + b\)。但这可能并不是最小整数解,所以我们需要调整为 \(x' = (x_0 \bmod b + b) \bmod b\);
  • 在 P2613 中,非负整数 \(a, b\) 的范围达到 \(10 ^{10001}\),所以需要结合快读一边读入一边取模;
  • 在 P2613 中,需要判断无解情况。

P1516 青蛙的约会

如果他们相遇,他们初始的位置坐标之差和跳的距离应该在模 \(l\) 下同余。

令 \(k\) 为跳的次数,则 \((n - m)k \equiv x - y \pmod l\)。

套模板即可,注意 \(n - m\) 和 \(x - y\) 可能为负数和无解情况。

P5656 【模板】二元一次不定方程 (exgcd)

模板简单,模板题不一定简单。因为此题细节很多,需要一一处理。

下面所有变量源于 Part 3 的应用 1。

那么我们先求最小值。首先,肯定的是用 Exgcd 求出一组特解 \(x_0, y_0\),那么显然 \(x, y\) 最小正整数解为 \(x_{\min} = (x_0 \bmod \Delta x + \Delta x) \bmod \Delta x\),\(y_{\min} = (y_0 \bmod \Delta y + \Delta y) \bmod \Delta y\)(读者可以简单证明)。

然后就可以求最大值了。最大值怎么求?很显然,对于 \(x_{\min}, y_{\min}\),有 \(ax_{\min} + by_{\max} = c\) 和 \(ax_{\max} + by_{\min} = c\)。那么 \(y_{\max} = \cfrac{c - ax_{\min}}{b}\),\(x_{\max} = \cfrac{c - by_{\min}}{a}\)。

你以为这就结束了?我们还没有处理没有正整数解的情况。当我们限制 \(x, y > 0\) 时,可以推导出 \(k\) 的取值范围,当 \(k\) 的下界大于上界时就没有正整数解。

Code:

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;

const int kMaxN = 2e5 + 10;

ll a, b, c, d, x, y, dx, dy, m, n, xmn, xmx, ymn, ymx;

ll Exgcd(ll a, ll b, ll &x, ll &y) {
  if (!b) {
    x = 1, y = 0;
    return a;
  }
  ll res = Exgcd(b, a % b, y, x);
  y -= a / b * x;
  return res;
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  int t;
  for (cin >> t; t; -- t) {
    cin >> a >> b >> c;
    d = Exgcd(a, b, x, y);
    if (c % d) {
      cout << "-1\n";
      continue;
    }
    dx = b / d, dy = a / d;
    x *= c / d, y *= c / d;
    m = ceil(1.0 * (1 - x) / dx), n = floor(1.0 * (y - 1) / dy);
    if (m > n) {
      cout << x + m * dx << ' ' << y - n * dy << '\n';
    } else {
      xmn = (x % dx + dx) % dx;
      if (!xmn) {
        xmn = dx;
      }
      ymn = (y % dy + dy) % dy;
      if (!ymn) {
        ymn = dy;
      }
      xmx = (c - ymn * b) / a, ymx = (c - xmn * a) / b;
      cout << n - m + 1 << ' ' << xmn << ' ' << ymn << ' ' << xmx << ' ' << ymx << '\n';
    }
  }
  return 0; 
}

乘法逆元

Part 0:前置知识

  • 基础数学(四则运算,符号定义等);
  • 费马小定理。

Part 1:乘法逆元的定义

1.1:为什么需要乘法逆元?

我们先来探讨一下,为什么需要乘法逆元?

因为,对于加减乘运算,我们都可以通过四则运算和模运算进行取模。但是,对于除法:\((a \div b) \bmod m \neq ((a \bmod m) \div (b \bmod m)) \bmod m\)。

1.2:乘法逆元的定义

对于整数 \(a\) 和模数 \(m\)(\(m > 1\)),如果存在整数 \(x\) 满足:

\[ax \equiv 1 \pmod p \]

则称 \(x\) 是 \(a\) 在模 \(m\) 意义下的乘法逆元,记作 \(a^{-1} \equiv x \pmod p\)。

注意:

这里的 \(a^{-1}\) 不直接等同于 \(\frac{1}{a}\),必须要在模运算下才可以完全等同。

1.3:逆元存在的条件

定理:\(a\) 在模 \(p\) 意义下有乘法逆元的充分必要条件是 \(\gcd(a, p) = 1\)。

证明:

  • 必要性:如果 \(\gcd(a, p) > 1\),假设存在逆元 \(x\),那么 \(ax = 1 + kp\)。由于 \(\gcd(a, p) \mid a\) 且 \(\gcd(a, p) \mid p\),所以 \(\gcd(a, b) \mid 1\),矛盾。
  • 充分性:如果 \(\gcd(a, p) = 1\),根据 Exgcd,存在 \(x, y \in \mathbb{Z}\) 使得 \(ax + py = 1\),于是 \(ax \equiv 1 \pmod p\)。

Part 2:乘法逆元的求法

2.1:Exgcd 求逆元

  • 原理:将同余方程转化为不定方程;
  • 优点:有判断无解的步骤,当 \(p\) 不是质数时也可使用;
  • 时间复杂度:单个 \(\mathcal{O}(\log x)\)。

我们将同余方程 \(a \cdot a^{-1} \equiv 1 \pmod p\) 转化成 \(ax + by = c\) 的形式:

\[\because a \cdot a^{-1} \equiv 1 \pmod p \therefore a \cdot a^{-1} = 1 + kp \therefore a \cdot a^{-1} + p \cdot (-k) = 1 \]

于是就可以使用 Exgcd 求解 \(a^{-1}\)。

Code:

cpp 复制代码
int Exgcd(int a, int b, int &x, int &y) {
  if (!b) {
    x = 1, y = 0;
    return a;
  }
  ll res = Exgcd(b, a % b, y, x);
  y -= a / b * x;
  return res;
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  cin >> a >> b;
  Exgcd(a, b, x, y);
  cout << (x % b + b) % b << '\n';
  return 0; 
}

2.2 费马小定理求逆元

  • 优点:代码简单;
  • 缺点:\(p\) 必须为质数;
  • 时间复杂度:\(\mathcal{O}(\log p)\)。

什么是费马小定理:

如果 \(p\) 是质数,且 \(\gcd(a, p) = 1\),那么

\[a^{p - 1} \equiv 1 \pmod p \]

现在我们开始求逆元:

由 \(a^{p - 1} \equiv 1 \pmod p\),得:

\[a \cdot a^{p - 2} \equiv 1 \pmod p \]

由于 \(a\) 是常数,所以当 \(p\) 是质数时,\(a\) 的逆元为:

\[a^{-1} \equiv a^{p - 2} \pmod p \]

于是,我们可以使用快速幂计算 \(a^{p - 2} \bmod p\)。

Code:

cpp 复制代码
ll Qpow(ll a, ll b, ll p) {
  ll ans = 1;
  for (; b; b >>= 1) {
    if (b & 1) {
      ans = ans * a % p;
    }
    a = a * a % p;
  }
  return ans;
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  cin >> n >> p;
  cout << Qpow(n, p - 2) << '\n';
  return 0; 
}

2.3 线性递推求逆元

  • 优点:可以一次性求 \(n\) 以内所有正整数模 \(p\) 的逆元;
  • 缺点:需满足 \(p\) 为质数且 \(n < p\);
  • 时间复杂度:\(\mathcal{O}(n)\)。

令 \(i\) 的逆元为 \(\text{inv}_i\),有如下递推式:

\[\text{inv}i = \left(p - \left\lfloor \frac{p}{i} \right\rfloor\right) \cdot \text{inv}{(p \bmod i)} \bmod p \]

边界条件为:\(\text{inv}_1 = 1\)。

证明:

令 \(p = ki + r\),其中 \(k = \left\lfloor \frac{p}{i} \right\rfloor\),\(r = p \bmod i\)。

在模 \(p\) 意义下:

\[ki + r \equiv 0 \pmod p \]

两边同乘 \(i^{-1}r^{-1}\),得:

\[kr^{-1} + i^{-1} \equiv 0 \pmod p \]

所以:

\[i^{-1} \equiv -kr^{-1} \pmod p \]

即:

\[\text{inv}_i \equiv (p - k)\text{inv}_r \pmod p \]

带入 \(k = \left\lfloor \frac{p}{i} \right\rfloor\),\(r = p \bmod i\),得:

\[\text{inv}i \equiv \left(p - \left\lfloor \frac{p}{i} \right\rfloor\right)\text{inv}{(p \bmod i)} \pmod p \]

得证。

Code:

cpp 复制代码
inv[0] = 0, inv[1] = 1;
for (int i = 2; i <= n; ++ i) {
  inv[i] = (p - p / i) * inv[p % i] % p;
} 

2.4 阶乘逆元的线性求法

  • 优点:可以一次性求 \(n\) 以内所有正整数的阶乘模 \(p\) 的逆元;
  • 条件:\(p\) 为质数;
  • 时间复杂度:\(\mathcal{O}(n)\)。

这种方法在组合数中用的比较多。

  1. 先递推计算阶乘:\(\text{fac}i = (\text{fac}{i - 1} \cdot i) \bmod p\)(边界条件:\(\text{fac}_1 = 1\));
  2. 用 Exgcd 或费马小定理求出 \(n!\) 的逆元 \(\text{invfac}_n\);
  3. 递推求其他阶乘逆元:\(\text{invfac}_i = ((i + 1) \cdot \text{invfac}_i) \bmod p\)。

证明:

因为 \(i! = \cfrac{(i + 1)!}{i + 1}\),所以 \((i!)^{-1} = (((i + 1)!)^{-1} \cdot (i + 1)) \bmod p\)。

Code:

cpp 复制代码
ll Qpow(ll a, ll b) {
  ll ans = 1;
  for (; b; b >>= 1) {
    if (b & 1) {
      ans = ans * a % kMod;
    }
    a = a * a % kMod;
  }
  return ans;
}

void Init() {
  fac[0] = fac[1] = 1;
  for (int i = 2; i <= n; ++ i) {
    fac[i] = fac[i - 1] * i % kMod;
  }
  inv[n] = Qpow(fac[n], kMod - 2);
  for (int i = n - 1; i; -- i) {
    inv[i] = inv[i + 1] * (i + 1) % kMod;
  }
}

Part 3:例题

其实单独考乘法逆元的题目很少,主要结合组合数学。

P1082 [NOIP 2012 提高组] 同余方程

这不是逆元板子题吗?

由于 \(b\) 未保证是质数,所以只能使用 Exgcd。

P3811 【模板】模意义下的乘法逆元

考虑使用费马小定理,但提交之后发现 TLE 了。

分析一下,费马小定理求逆元单次是 \(\mathcal{O}(\log p)\) 的,总体就是 \(\mathcal{O}(n \log p)\),在本题数据范围下超时。

所以必须使用线性递推求逆元,时间复杂度 \(\mathcal{O}(n)\)。

P2265 路边的水沟

简单分析可得,右下角的闸门到左上角的闸门的总共有 \(n + m\) 次移动次数。

而我们必须向上走 \(n\) 次,向右走 \(m\) 次。

答案就是在 \(n + m\) 次移动中选 \(n\) 个向上的方案数,答案即为 \(C^{n + m}_m\)。

使用阶乘逆元的线性求法即可。

P5431 【模板】模意义下的乘法逆元 2

由于:

\[\frac{1}{\prod_{j = 1}^i a_j} \cdot a_i \equiv \frac{1}{\prod_{j = 1}^{i - 1} a_j} \pmod p \]

令 \(\text{mul}i = \prod{j = 1}^i a_j\),则:

\[(\text{mul}{i - 1})^{-1} \equiv (\text{mul}{i})^{-1} \cdot a_i \pmod p \]

接下来求 \(a_i\) 的逆元:

因为:

\[\prod_{j = 1}^{i - 1} a_j \cdot \frac{1}{\prod_{j = 1}^{i} a_j} \equiv \frac{1}{a_i} \pmod p \]

所以:

\[(a_i)^{-1} \equiv i \cdot \text{mul}i \cdot \text{mul}{i - 1} \pmod p \]

别忘了,原式还有一个 \(k^i\),一边算一边乘即可。

本题需要配合快读使用。

Code:

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;

const int kMaxN = 5e6 + 10;

ll n, p, k, a[kMaxN], mul[kMaxN], inv[kMaxN], ans;

ll R() {
  ll x = 0, f = 1;
  char ch = getchar();
  while (!isdigit(ch)) {
    if (ch == '-') {
      f = -1;
    }
    ch = getchar();
  }
  while (isdigit(ch)) {
    x = x * 10 + (ch ^ 48);
    ch = getchar();
  }
  return x * f;
}

ll Qpow(ll a, ll b) {
  ll ans = 1;
  for (; b; b >>= 1) {
    if (b & 1) {
      ans = ans * a % p;
    }
    a = a * a % p;
  }
  return ans;
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  n = R(), p = R(), k = R(), mul[0] = 1;
  for (int i = 1; i <= n; ++ i) {
    a[i] = R();
    mul[i] = mul[i - 1] * a[i] % p;
  }
  inv[n] = Qpow(mul[n], p - 2);
  for (int i = n - 1; i; -- i) {
    inv[i] = inv[i + 1] * a[i + 1] % p;
  }
  for (int i = n; i; -- i) {
    ans = (ans + (inv[i] * mul[i - 1]) % p) % p * k % p;
  }
  cout << ans << '\n';
  return 0;
}

欧拉函数

Part 0:前置知识

  • 基础数学(四则运算,符号定义等)。
  • 最大公约数(gcd);
  • 互质。

Part 1:欧拉函数的定义

在信息学中,我们常常要解决互质类问题或处理大指数取模运算,而欧拉定理和欧拉函数就是一个好帮手。

定义:欧拉函数 \(\varphi(n)\) 表示小于等于 \(n\) 的正整数中与 \(n\) 互质的数的个数,即 \(\varphi(n) = |\{k \in \mathbb{N^+} \mid 1 \le k \le n,\ \gcd(k, n) = 1\}|\)。

欧拉函数有一个核心公式,此公式用处较多。

若 \(n = p_1^{k_1}p_2^{k_2} \cdots p_n^{k_n}\),则:

\[\varphi(n) = n \cdot \prod_{i = 1}^m \left(1 - \frac{1}{p_i}\right) \]

展开写就是:

\[\varphi(n) = p_1^{k_1 - 1}(p_1 - 1)p_2^{k_2 - 1}(p_2 - 1) \cdots p_n^{k_n - 1}(p_n - 1) \]

此公式可以使用下文的性质 1 证明。

Part 2:欧拉函数的性质

2.1:性质 1

对于任意质数 \(p\),有 \(\varphi(p) = p - 1\),对于 \(p\) 的幂次 \(p^k\)(\(k \geq 1\)),有 \(\varphi(p^k) = p^k - p^{k - 1} = p^{k - 1}(p - 1)\)。

证明:

在 \(1, 2, \cdots, p\) 中,不与 \(p\) 互质的数只有 \(p\) 本身,故 \(\varphi(p) = p - 1\)。

在 \(1, 2, \cdots, p^k\) 中,只有 \(p\) 的倍数不与 \(p\) 互质,即 \(p, 2p, 3p, \cdots, p^{k}\)。由于这些数有 \(p^{k - 1}\) 个,所以 \(\varphi(p^k) = p^k - p^{k - 1} = p^{k - 1}(p - 1)\)。

2.2:性质 2

若 \(a \mid x\),则 \(\varphi(ax) = a \cdot \varphi(x)\)。

此性质证明计算量较大。

由 \(a \mid x\),设 \(x = \prod_{i = 1}^n p_i^{k_i}\),\(a = \prod_{i = 1}^n p_i^{s_i}\)(\(0 \leq s_i \leq k_i\))。

那么:

\[ax = \prod_{i=1}^n p_i^{k_i+s_i} \]

使用用欧拉函数公式:

\[\varphi(ax) = \prod_{i = 1}^n \varphi(p_i^{k_i + s_i}) = \prod_{i = 1}^n \bigl(p_i^{k_i + s_i} - p_i^{k_i + s_i - 1}\bigr) \]

\[a \cdot \varphi(x) = \left(\prod_{i = 1}^n p_i^{s_i}\right) \cdot \prod_{i = 1}^n \bigl(p_i^{k_i} - p_i^{k_i - 1}\bigr) = \prod_{i = 1}^n p_i^{s_i} \bigl(p_i^{k_i} - p_i^{k_i - 1}\bigr) \]

对每一项:

\[p_i^{s_i} \bigl(p_i^{k_i} - p_i^{k_i - 1}\bigr) = p_i^{k_i + s_i} - p_i^{k_i + s_i - 1} = \varphi(p_i^{k_i + s_i})\]

因此 \(\varphi(ax) = a \cdot \varphi(x)\)。

2.3:性质 3

欧拉函数是积性函数。

即,对于任意满足 \(\gcd(a, b) = 1\) 的整数 \(a, b\),都有 \(\varphi(ab) = \varphi(a)\varphi(b)\)。

特别地,当 \(n \bmod 2 = 1\) 时,\(\varphi(2n) = \varphi(n)\)。

此性质证明需要用到作者的知识盲区,作者太弱所以就不写上去了。

Part 3:欧拉函数的计算

3.1:暴力计算欧拉函数

欧拉函数公式,枚举每一个质因子然后计算答案即可。

时间复杂度:单个 \(\mathcal{O}(\log n)\)。

Code:

cpp 复制代码
ll Phi(ll n) {
  ll ans = n;
  for (ll i = 2; i * i <= n; ++ i) {
    if (!(n % i)) {
      ans -= ans / i;
      for (; !(n % i); n /= i);
    }
  }
  if (n > 1) {
    ans -= ans / n;
  }
  return ans;
}

3.2 线性筛求解欧拉函数

我们借助线性筛来求解欧拉函数,这个求法主要依赖于欧拉函数的性质。先放代码,再讲原理。

Code:

cpp 复制代码
const int kMaxN = 1e6 + 10;

ll p[kMaxN], phi[kMaxN], cnt;
bool vis[kMaxN];

void Init() {
  vis[1] = phi[1] = 1;
  for (ll i = 2; i < kMaxN; ++ i) {
    if (!vis[i]) {
      phi[i] = i - 1;
      p[++ cnt] = i;
    }
    for (ll j = 1; j <= cnt && i * p[j] < kMaxN; ++ j) {
      vis[i * p[j]] = 1;
      phi[i * p[j]] = phi[i] * (p[j] - 1);
      if (!(i % p[j])) {
        phi[i * p[j]] = phi[i] * p[j];
        break;
      }
    }
  }
}

令 \(p_i\) 表示第 \(i\) 个质数,\(\text{vis}_i = 1\) 表示 \(i\) 是合数。

考虑枚举 \(2\) 至 \(N\) 的所有整数。

如果 \(\text{vis}_i = 1\),就证明 \(i\) 未被筛出,即 \(i\) 是质数,那么 \(\varphi(i) = i - 1\)。

然后,我们就开始筛除 \(i\) 的倍数。如果 \(i \mid p_j\),那么根据性质 2,得到 \(\varphi(i \cdot p_j) = \varphi(i) \cdot p_j\)。否则,我们根据性质 3(积性函数的性质),可以得到 \(\varphi(i \cdot p_j) = \varphi(i) \cdot (p_j - 1)\)。

时间复杂度 \(\mathcal{O}(N)\)。

Part 4:例题

UVA11327 Enumerating Rational Numbers

解读一下代码,就可以发现对于每一个 \(d\),满足条件的分数个数就是 \(\varphi(d)\)。

于是尝试枚举 \(d\),每一次将个数与 \(\varphi(d)\) 相加。如果发现超过 \(k\) 了,那就证明答案的 \(d\) 一定是当前的 \(d\),此时枚举 \(n\) 即可。

由于 \(d \le 2 \times 10^5\)(由样例发现),线性筛预处理即可。

CF1295D Same GCDs

首先对式子做一次辗转相除,得:

\[\gcd(a + x,\ m) = \gcd((a + x) \bmod m,\ m) \]

显然题目可以转化为求有多少 \(x\) 满足 \(\gcd(x, m) = \gcd(a, m)\)。

令 \(\gcd(x, m) = \gcd(a, m) = d\),则 \(\gcd\left(\cfrac{x}{d}, \cfrac{m}{d}\right) = 1\)。

现在再次转换题目:求与 \(\cfrac{m}{d}\) 互质的数的个数。显然就是欧拉函数,答案为 \(\varphi \left(\cfrac{m}{d}\right)\)。

P1891 疯狂 LCM

遇到这种题,无脑暴力推式子:

\[\sum_{i = 1}^n \text{lcm}(i, n) = \sum_{i = 1}^n \cfrac{i \cdot n}{\gcd(i, n)} = n \sum_{i = 1}^n \cfrac{i}{\gcd(i, n)} = n \sum_{d \mid n} \sum_{i = 1}^n \frac{i}{d} \cdot [\gcd(i, n) = d] = n \sum_{d \mid n} \sum_{i = 1}^n \frac{i}{d} \cdot \left[\gcd\left(\frac{i}{d}, \frac{n}{d}\right) = 1\right] = n \sum_{d \mid n} \sum_{i = 1}^{\frac{n}{d}} i \cdot \left[\gcd\left(i, \frac{n}{d}\right) = 1\right] = n \sum_{d \mid n} \sum_{i = 1}^{d} i \cdot [\gcd(i, d) = 1] \]

分析上式,目前的目标就是求 \(\sum_{i = 1}^{d} i \cdot [\gcd(i, d) = 1]\)。

假设你欧拉函数掌握的不是特别好,我们来找一下规律。假设 \(n = 8\),那么 \(\gcd(i, n) = 8\) 的有 \(1, 3, 5, 7\)。这时细心的人可以发现 \(1 + 7 = 3 + 5 = n = 8\)。

没错!对于 \(\gcd(i, d) = 1\),也存在 \(\gcd(d - i,\ d) = 1\)(证明很简单)!

那么:

\[\sum_{i = 1}^{d} i \cdot [\gcd(i, d) = 1] = d \cdot \frac{\varphi(d)}{2} \]

答案即为:

\[n \sum_{d \mid n} d \cdot \frac{\varphi(d)}{2} \]

那么,我们先 \(\mathcal{O}(N)\) 预处理欧拉函数,然后 \(\mathcal{O}(N \log N)\) 预处理每一个 \(d\) 的 \(f(d) = \sum_{d \mid n} d \cdot \frac{\varphi(d)}{2}\)(可以使用 DP 处理)。最后,询问时直接输出 \(n \cdot f(n)\) 即可。

Code:

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;

const int kMaxN = 1e6 + 10;

ll n, phi[kMaxN], p[kMaxN], f[kMaxN], cnt;
bool vis[kMaxN];

void Init() {
  vis[1] = phi[1] = 1;
  for (ll i = 2; i <= 1e6; ++ i) {
    if (!vis[i]) {
      phi[i] = i - 1;
      p[++ cnt] = i;
    }
    for (ll j = 1; j <= cnt && i * p[j] <= 1e6; ++ j) {
      vis[i * p[j]] = 1;
      phi[i * p[j]] = phi[i] * (p[j] - 1);
      if (!(i % p[j])) {
        phi[i * p[j]] = phi[i] * p[j];
        break;
      }
    }
  }
  for (ll i = 1; i <= 1e6; ++ i) {
    for (ll j = i; j <= 1e6; j += i) {
      f[j] += (phi[i] * i + 1) / 2;
    }
  }
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  Init();
  int t;
  for (cin >> t; t; -- t) {
    cin >> n;
    cout << n * f[n] << '\n';
  }
  return 0;
}

中国剩余定理(CRT)

Part 0:前置知识

  • 基础数学(四则运算,符号定义等)。
  • 互质;
  • 同余;
  • 逆元;
  • 裴蜀定理。

Part 1:中国剩余定理是什么?

中国剩余定理(Chinese Remainder Theorem,CRT),又称孙子定理,是数论中最具代表性的定理之一,最早记载于我国南北朝时期的数学著作《孙子算经》,其中经典的"物不知数"问题便是其雏形:"今有物不知其数,三三数之剩二,五五数之剩三,七七数之剩二,问物几何?"

中国剩余定理一般用来求解以下一次线性同余方程组:

\[\begin{cases} x \equiv b_1 \pmod{m_1} \\ x \equiv b_2 \pmod{m_2} \\ \cdots \\ x \equiv b_n \pmod{m_n} \end{cases} \]

其中对于 \(i \neq j\),有 \(\gcd(m_i, m_j) = 1\)。

Part 2:同余方程组的求法

我们考虑使用余数的可加性,那么题目转化为求出 \(n\) 个方程组的解(\(i \in [1, n]\)):

\[\begin{cases} x_i \equiv 0 \pmod{m_1} \\ x_i \equiv 0 \pmod{m_2} \\ \cdots \\ x_i \equiv b_i \pmod{m_i} \\ \cdots \\ x_i \equiv 0 \pmod{m_n} \end{cases} \]

那么 \(x = \sum_{i = 1}^n x_i\)。

进一步转化,其实就是求这 \(n\) 个方程组的解:

\[\begin{cases} y_i \equiv 0 \pmod{m_1} \\ y_i \equiv 0 \pmod{m_2} \\ \cdots \\ y_i \equiv 1 \pmod{m_i} \\ \cdots \\ y_i \equiv 0 \pmod{m_n} \end{cases} \]

由余数的可乘性,得 \(x_i = y_ib_i\)。由于对于 \(j \neq i\),有 \(y_i \equiv 0 \pmod{m_j}\),那么 \(y_i\) 一定可以表示为 \(k_i \cdot \prod_{j = 1}^n m_i[j \neq i]\)(\(k \in \mathbb{N^+}\))。

又 \(y_i \equiv 0 \pmod{m_1}\),那么 \(k_i \cdot \prod_{j = 1}^n m_i[j \neq i] \equiv 1 \pmod {m_i}\)。观察式子,发现与逆元的定义相同。那么求 \(y_i\) 就转为了求 \(\prod_{j = 1}^n m_j[j \neq i] \bmod m_i\) 的逆元。这时,就可以拿 Exgcd 求解了。

求出 \(n\) 个方程组的解后,就得到了原方程的整数解。但因为题目要求最小非负整数解,所以还要再进行对乘积的取模。

Code:

cpp 复制代码
ll Exgcd(ll a, ll b, ll &x, ll &y) {
  if (!b) {
    x = 1, y = 0;
    return a;
  }
  ll d = Exgcd(b, a % b, y, x);
  y -= a / b * x;
  return d;
}

ll CRT() {
  ll mul = 1, ans = 0;
  for (int i = 1; i <= n; ++ i) {
    mul *= b[i];
  }
  for (ll i = 1, k, x, y; i <= n; ++ i) {
    k = mul / b[i];
    Exgcd(k, b[i], x, y);
    ans = (ans + k * b[i] * x % mul) % mul;
  }
  return (ans % mul + mul) % mul;
}

Part 3:扩展中国剩余定理(ExCRT)

由于中国剩余定理的局限性太强(两两互质),所以我们需要扩展中国剩余定理。

在信息学中,CRT 与 ExCRT 复杂度一致。所以,只要学习扩展中国剩余定理就可以覆盖 CRT 的问题。

考虑如下方程组:

\[\begin{cases} x \equiv b_1 \pmod{m_1} \\ x \equiv b_2 \pmod{m_2} \\ \cdots \\ x \equiv b_n \pmod{m_n} \end{cases} \]

其中 \(m_i\) 是任意正整数。

首先,我们将式子等价转换。

对于 \(x \equiv b_1 \pmod{m_1}\) 和 \(x \equiv b_2 \pmod{m_2}\),则有:

\[\begin{cases} x = k_1m_1 + b_1 \\ x = k_2m_2 + b_2\end{cases} \]

于是:

\[k_1m_1 + b_1 = k_2m_2 + b_2 \]

移项,得:

\[k_1m_1 - k_2m_2 = b_2 - b_1 \]

观察此式子,发现形态与线性不定方程相同,考虑使用扩展欧几里得(Exgcd)求解。

令 \(d = \gcd(m_1, m_2)\)。

首先,根据裴蜀定理:若 \(d \nmid (b_2 - b_1)\),方程无解。否则,我们可以使用 Exgcd 求解。

设通过 Exgcd 得到:

\[m_1 \cdot p + m_2 \cdot q = d \]

两边同乘 \(\frac{b_2 - b_1}{d}\):

\[m_1 \cdot \frac{p(b_2 - b_1)}{d} + m_2 \cdot \frac{q(b_2 - b_1)}{d} = b_2 - b_1 \]

得到一组特解:

\[\begin{cases} k_1 = \cfrac{p(b_2 - b_1)}{d} \\ k_2 = -\cfrac{q(b_2 - b_1)}{d} \end{cases} \]

那么,原方程组的通解为:

\[x = b_1 + k_1m_1 + t \cdot \text{lcm}(m_1, m_2) = b_1 + k_1m_1 + t \cdot \frac{m_1m_2}{d} \]

然后我们考虑使用数学归纳法。

设前 \(k - 1\) 个方程的一个特解为 \(x'\),则通解为 \(x'+ t \cdot \text{lcm}(m_1, m_2, \cdots, m_{k - 1})\)(\(t \in \mathbb{Z^+}\))。对于第 \(k\) 个方程 \(x \equiv b_k \pmod{m_k}\),将通解代入第 \(k\) 个方程的 \(x\):

\[x_1 + t \cdot \text{lcm}(m_1, m_2, \cdots, m_{k - 1}) \equiv b_k \pmod{m_k} \]

移项,得:

\[t \cdot \text{lcm}(m_1, m_2, \cdots, m_{k - 1}) \equiv b_k - x_1 \pmod{m_k} \]

将同余方程转成一次不定方程,得:

\[\text{lcm}(m_1, m_2, \cdots, m_{k - 1}) \cdot t + m_k \cdot y = b_k - x_1 \]

若该方程无解,则说明前 \(k\) 个方程无解,退出。否则,更新特解 \(x_1\) 和 \(\text{lcm}\)。

Code:

cpp 复制代码
ll Exgcd(ll a, ll b, ll &x, ll &y) {
  if (!b) {
    x = 1, y = 0;
    return a;
  }
  ll res = Exgcd(b, a % b, y, x);
  y -= a / b * x;
  return res;
}

ll Qmul(ll a, ll b, ll p) {
  ll ans = 0;
  for (; b; b >>= 1) {
    if (b & 1) {
      ans = (ans + a) % p;
    }
    a = (a + a) % p;
  }
  return ans;
}

ll ExCRT() {
  ans = b[1], lcm = m[1];
  for (ll i = 2, d, x, y, k; i <= n; ++ i) {
    b[i] = ((b[i] - ans) % m[i] + m[i]) % m[i];
    d = Exgcd(lcm, m[i], x, y);
    if (b[i] % d) {
      ans = -1;
      break;
    }
    k = Qmul(x, b[i] / d, m[i]);
    ans += k * lcm, lcm = lcm / d * m[i];
    ans = (ans % lcm + lcm) % lcm;
  }
  return ans;
}

Part 4:例题

P1495 【模板】中国剩余定理(CRT)/ P4777 【模板】扩展中国剩余定理(EXCRT)

板子题,套模板即可。

注意 P1495 需要 __int128

P3868 [TJOI2009] 猜数字

题目中的式子很难看,那我们就把它写成同余方程组:\(n - a_i \equiv 0 \pmod{b_i}\)。

因为 \(a \equiv b \pmod m\) 时,有 \(a + c \equiv b + c \pmod m\)。所以题目中的式子就转化成 \(n \equiv a_i \pmod{b_i}\)。

然后就是模板了。

CF687B Remainders Game

此题思维性较强。

我们由题目可以知道,Arya 可以从 Pari 手中得知 \(x \bmod c_i\) 的值。也就是,令第 \(i\) 次询问的结果是 \(b_i\),那么 Alice 可以得到如下式子:

\[\begin{cases} x \equiv a_1 \pmod{c_1} \\ x \equiv a_2 \pmod{c_2} \\ \cdots \\ x \equiv a_n \pmod{c_n} \end{cases} \]

这时你会发现,这个方程组与 ExCRT 一摸一样!那么,我们结合 ExCRT,可以发现,通过任意的:

\[\begin{cases} x \equiv a_i \pmod{c_i} \\ x \equiv a_{i + 1} \pmod{c_{i + 1}}\end{cases} \]

可以得到 \(x \bmod \text{lcm}(c_i, c_{i + 1})\) 的值!也就是,我们只需要知道 \(c_i, c_{i + 1}\) 和 \(x \bmod c_i\),\(x \bmod c_{i + 1}\),就可以得出 \(x \bmod \text{lcm}(c_i, c_{i + 1})\) 的值。

因此,我们只需要求出 \(\text{lcm}(c_1, c_2, \cdots, c_n)\) 的值,然后判断是否有 \(k \mid \text{lcm}(c_1, c_2, \cdots, c_n)\) 即可。

为什么呢?因为当 \(\text{lcm}(c_1, c_2, \cdots, c_n)\) 满足上述条件时,我们可以得到 \(x \bmod tk\)(\(t\) 为任意正整数)的值。那么,根据同余的性质,便可以得到 \(x \bmod k\)。

欧拉定理

Part 0:前置知识

  • 基础数学(四则运算,符号定义等)。
  • 互质。

Part 1:欧拉定理的内容

1.1:欧拉定理

若正整数 \(a, n\) 满足 \(\gcd(a, n) = 1\),那么有:

\[a^{\varphi(n)} \equiv 1 \pmod n \]

当 \(n\) 为质数 \(p\) 时,\(\varphi(p) = p - 1\),欧拉定理退化为费马小定理:

\[a^{p - 1} \equiv 1 \pmod p \]

1.2:证明

接下来,我们证明欧拉定理。

但是,在证明之前,我们需要先了解一个关键的观察:

取 \(n = 8\),\(a = 3\),将与 \(n\) 互质的数乘以 \(a\) 在对 \(n\) 取模:

\[\begin{cases} 3 \times 1 \equiv 3 \pmod 8 \\ 3 \times 3 \equiv 1 \pmod 8 \\ 3 \times 5 \equiv 7 \pmod 8 \\ 3 \times 7 \equiv 5 \pmod 8\end{cases} \]

我们发现,得到的余数是 \(3, 1, 7, 5\),恰好是与 \(n\) 互质的数 \(1, 3, 5, 7\) 的一个重新排列。

这不是巧合,这是一个普遍规律。

那么,接下来我们开始证明。

设所有小于等于 \(n\) 且与 \(n\) 互质的正整数为:

\[r_1, r_2, \cdots, r_{k} \]

由欧拉函数,得 \(k = \varphi(n)\)。

考虑将这些数分别乘上 \(a\),然后对 \(n\) 取模:

\[ar_1 \bmod n,\ ar_2 \bmod n,\ \cdots,\ ar_{\varphi(n)} \bmod n \]

接下来,我们需要证明三个性质。

性质 1:\(\gcd(ar_i, n) = 1\)。

因为 \(\gcd(a, n) = \gcd(r_i, n) = 1\),那么根据互质的性质,得 \(\gcd(ar_i, n) = 1\)。

性质 2:对于 \(i \neq j\),有 \(ar_i \not \equiv ar_j \pmod n\)。

假设 \(ar_i \equiv ar_j \pmod n\),即 \(n \mid a(r_i - r_j)\)。

因为 \(\gcd(a, n) = 1\),所以 \(n \mid (r_i - r_j)\),即 \(r_i \equiv r_j \pmod n\)。

又 \(r_i, r_j \in [1, n]\) 且 \(\gcd(r_i, n) = \gcd(r_j, n) = 1\),所以 \(r_i = r_j\)。

因为 \(r_i, r_j\) 互不相同,所以 \(i = j\)。

与 \(i \neq j\) 矛盾,原命题得证。

性质 3:\((ar_1 \bmod n) \in [1, n)\)。

将命题转化成证明 \(ar_1 \bmod n \neq 0\),即 \(ar_1 \not \equiv 0 \pmod n\)。

假设 \(ar_1 \equiv 0 \pmod n\),那么 \(n \mid ar_i\),但 \(\gcd(a, n) = \gcd(r_i, n) = 1\)。

矛盾,原命题得证。

三个性质证明完后,我们会发现一个关键结论。

现在我们有两个集合:

  • 原始集合 \(S = \{r_1, r_2, \cdots, r_{\varphi(n)}\}\);
  • 乘积集合 \(S' = \{ar_1 \bmod n,\ ar_2 \bmod n,\ \cdots,\ ar_{\varphi(n)} \bmod n\}\)。

因为乘积集合也具有以下三个性质:

  • \(\gcd(ar_i, n) = 1\);
  • 对于 \(i \neq j\),有 \(ar_i \not \equiv ar_j \pmod n\);
  • \((ar_1 \bmod n) \in [1, n)\)。

这表明一个关键结论:\(S = S'\)。

既然 \(S = S'\),那么元素的乘积应该相等,那么推导出:

\[\prod_{i = 1}^{\varphi(n)} r_i \equiv \prod_{i = 1}^{\varphi(n)} ar_i \pmod n \]

将 \(\prod_{i = 1}^{\varphi(n)} ar_i \bmod n\) 展开:

\[\prod_{i = 1}^{\varphi(n)} ar_i \bmod n = a^{\varphi(n)} \cdot \prod_{i = 1}^{\varphi(n)} r_i \bmod n \]

所以有:

\[a^{\varphi(n)} \cdot \prod_{i = 1}^{\varphi(n)} r_i \equiv \prod_{i = 1}^{\varphi(n)} r_i \pmod n \]

由于原始集合的乘积与 \(n\) 互质,故原始集合的乘积在模 \(n\) 下有乘法逆元,在式子两边同乘它的逆元:

\[a^{\varphi(n)} \equiv 1\pmod n \]

证毕。

Part 2:欧拉定理的应用

2.1:求乘法逆元

如果 \(\gcd(a, n) = 1\),则 \(a\) 在模 \(n\) 意义下的乘法逆元为:

\[a^{-1} \equiv a^{\varphi(n) - 1} \pmod n \]

证明:

\[a \cdot a^{\varphi(n) - 1} = a^{\varphi(n)} \equiv 1 \pmod n \]

注意:

此处实际上比费马小定理求逆元的应用更广。

2.2:指数降幂

对于 \(a^b \bmod n\),如果 \(\gcd(a, n) = 1\),则:

\[a^b \equiv a^{b \bmod \varphi(n)} \pmod n \]

证明:

令 \(b = q \cdot \varphi(n) + r\)(其中 \(r \in [0, \varphi(n))\)),则:

\[a^b = a^{q \cdot \varphi(n) + r} = (a^{\varphi(n)})^q \cdot a^r \equiv 1^q \cdot a^r \equiv a^r \pmod n \]

Part 3:扩展欧拉定理

对于任意正整数 \(n\) 和任意非负整数 \(b\),有:

\[a^b \equiv \begin{cases} a^{b \bmod \varphi(n)} & \text{if } \gcd(a, n) = 1 \\ a^b & \text{if } \gcd(a, n) \neq 1,\ b < \varphi(n) \\ a^{b \bmod \varphi(n) + \varphi(n)} & \text{if } \gcd(a, n) \neq 1,\ b \ge \varphi(n) \end{cases} \pmod n \]

注意:不需要满足 \(\gcd(a, n) = 1\)。

扩展欧拉定理给出了降幂更加广泛的形式。

Code:

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;

const int kMaxN = 1e6 + 10;

ll a, b, m, kMod;
bool ok;

ll R() {
  ll k = 0, f = 1;
  char c = getchar();
  while (c < '0' || c > '9') {
    (c == '-') && (f = -1);
    c = getchar();
  }
  while (c >= '0' && c <= '9') {
    k = k * 10 + c - '0';
    if (k >= kMod) {
      ok = 1;
    }
    k %= kMod;
    c = getchar();
  }
  return k * f;
}

ll Phi(ll n) {
  ll ans = n;
  for (ll i = 2; i * i <= n; ++ i) {
    if (!(n % i)) {
      ans -= ans / i;
      for (; !(n % i); n /= i);
    }
  }
  if (n > 1) {
    ans -= ans / n;
  }
  return ans;
}

ll Qpow(ll b, ll p) {
  ll ans = 1;
  for (; p; p >>= 1) {
    if (p & 1) {
      ans = ans * b % m;
    }
    b = b * b % m;
  }
  return ans;
}

int main() {
  cin >> a >> m;
  kMod = Phi(m), b = R();
  if (ok) {
    b += kMod;
  }
  cout << Qpow(a, b) << '\n';
  return 0;
}

Part 4:例题

P5091 【模板】扩展欧拉定理

板子题,套模板即可。

P4139 上帝与集合的正确用法

上面的文字很多,不用在意。我们直接将这个数列转化成求:

\[2^{2^{2^{2^{\cdots}}}} \bmod p \]

也就是 \(2\) 的无限幂塔。

考虑使用扩展欧拉定理。可以证明,指数满足 \(b \ge varphi(p)\)。

于是,考虑使用递归的形式,直到边界条件 \(p = 1\),此时式子的值是 \(0\)。

P2350 [HAOI2012] 外星人

由于一般的题目不单独考欧拉定理,所以此题严格上归于"欧拉函数"。

题目的意思就是:给定 \(n\),每次将 \(n\) 变为 \(\varphi(n)\),求几次操作后 \(n = 1\)。

首先,人人皆知,只有 \(n = 1\) 或 \(2\) 时才有 \(\varphi(n) = 1\)。

接下来,根据题目中给的提示,我们有:

\[\varphi \left(\prod_{i = 1}^m p_i^{q_i} \right) = \prod_{i = 1}^m p_i^{q_i - 1}(p_i - 1) \]

观察到,式子中有一个 \((p_i - 1)\)。那么,每一次会把上一次操作的答案中的最多一个 \(2\) 变为 \(1\)。

所以,我们只需要求出 \(n\) 的每一个质因子会产生多少个 \(2\) 即可。令 \(n = i\) 时答案为 \(f_i\),我们分情况讨论:

  • 当 \(i\) 是质数时,显然有递推式 \(f_i = f_{i - 1}\);
  • 当 \(i\) 不是质数时,令 \(i = p \cdot q\)。那么根据乘积的性质,可以得到 \(f_i = f_p + f_q\)。

你以为这就结束了?错!如果当原数 \(n\) 中没有因子 \(2\) 时,第一次操作不会把任何 \(2\) 变为 \(1\)。

所以,当原数 \(n\) 中没有因子 \(2\) 时,答案需减少 \(1\)。

Code:

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;

const int kMaxN = 1e5 + 10;

ll m, f[kMaxN], p[kMaxN], q[kMaxN], cnt, ans;
bool isp[kMaxN];

void Init() {
  f[1] = 1;
  for (ll i = 2; i < kMaxN; ++ i) {
    if (!isp[i]) {
      p[++ cnt] = i, f[i] = f[i - 1];
    }
    for (ll j = 1; j <= cnt && p[j] * i < kMaxN; ++ j) {
      isp[p[j] * i] = 1, f[p[j] * i] = f[p[j]] + f[i];
      if (!(i % p[j])) {
        break;
      }
    }
  }
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  int t;
  Init();
  for (cin >> t; t; -- t) {
    cin >> m, ans = 1;
    for (int i = 1; i <= m; ++ i) {
      cin >> p[i] >> q[i];
      ans += f[p[i]] * q[i] + (p[i] % 2? 0 : -1);
    }
    cout << ans << '\n';
  }
  return 0;
}

小步大步算法(BSGS)

Part 0:前置知识

  • 基础数学(四则运算,符号定义等)。
  • 模运算;
  • 欧拉定理;
  • 离散对数(可以不掌握)。

Part 1:小步大步算法

1.1:算法解决的问题

小步大步算法(Baby Step Giant Step)一般解决如下问题:

给定一个质数 \(p\) 和整数 \(a, b\)(其中 \(a \in (0, p)\),\(b \in [0, p)\)),求最小的非负整数 \(x\),满足:

\[a^x \equiv b \pmod p \]

或者报告无解。

1.2:算法思想

此算法的思想是根号类思想,结合 Meet in middle 的思想。

令 \(t = \lceil\sqrt{p}\rceil\),用 \(t\) 表示 \(x\),则 \(x = At - B\),那么:

\[a^x = a^{At - B} \equiv b \pmod p \]

两边同乘 \(a^B\):

\[a^{At} \equiv b \cdot a^{B} \pmod p \]

考虑枚举 \(B\),范围是 \([0, t - 1]\),将所有的 \(b \cdot a^{B}\) 用 Hash 表存储。

再考虑枚举 \(A\),范围是 \([1, t]\),计算 \(A\) 个 \(a^t\) 的积。如果在 Hash 表中出现过,说明找到了对应的 \(A, B\)。

此时,\(x = At - B\) 即为答案。

总时间复杂度 \(\mathcal{O}(t)\),即 \(\mathcal{O}(\sqrt{p})\)。

Code:

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;

const int kMaxN = 1e5 + 10;

ll p, b, n;

ll Qpow(ll a, ll b, ll p) {
  ll ans = 1;
  for (; b; b >>= 1) {
    if (b & 1) {
      ans = ans * a % p;
    }
    a = a * a % p;
  }
  return ans;
}

ll BSGS(ll p, ll b, ll n) {
  unordered_map<ll, ll> mp;
  ll t = sqrt(p) + 1;
  for (ll i = 1, mul = (n * b) % p; i <= t; ++ i, mul = (mul * b) % p) {
    mp[mul % p] = i;
  } 
  ll pw = Qpow(b, t, p);
  for (ll i = 1, mul = pw; i <= t; ++ i, mul = (mul * pw) % p) {
    if (mp[mul]) {
      return t * i - mp[mul];
    }
  }
  return -1;
} 

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  cin >> p >> b >> n;
  ll ans = BSGS(p, b, n);
  if (ans == -1) {
    cout << "no solution\n";
  } else {
    cout << ans << '\n';
  }
  return 0;
}

Part 2:扩展小步大步算法

在扩展小步大步算法(exBSGS)中,不要求 \(\gcd(a, p) = 1\)。

那么我们开始推导。

首先,我们尝试将同余方程写成不定方程:

\[a \cdot a^{x - 1} + py = b \]

根据裴蜀定理,当 \(\gcd(a, p) \nmid b\) 时方程无整数解。否则,我们令 \(d = \gcd(a, b)\),将不定方程两边同除 \(d\):

\[\frac{a}{d} \cdot a^{x - 1} + \frac{p}{d} \cdot y = \frac{b}{d} \]

再将不定方程写成同余方程的形式:

\[\frac{a}{d} \cdot a^{x - 1} \equiv \frac{b}{d} \pmod{\frac{p}{d}} \]

两边同乘 \(\frac{a}{d}\) 在模 \(\frac{p}{d}\) 下的逆元:

\[a^{x - 1} \equiv \frac{b}{d} \cdot \left(\frac{a}{d}\right)^{-1} \pmod{\frac{p}{d}} \]

我们发现,这跟 BSGS 的形式完全相同。

所以,我们检查 \(\gcd(a, \frac{p}{d})\) 是否为 \(1\)。如果满足,则使用 BSGS 求出 \(x - 1\)。否则,重复执行上述操作,直到满足条件或得到无解。

Code:

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;

const int kMaxN = 2e5 + 10;

ll BSGS(ll a, ll p, ll b, ll pw) {
  unordered_map<ll, ll> mp;
  ll t = sqrt(p) + 1, mul = 1;
  for (ll i = 1; i <= t; ++ i, mul = mul * a % p) {
    mp[mul * b % p] = i;
  }
  for (ll i = 1, k = mul, mul = pw; i <= t + 1; ++ i, mul = mul * k % p) {
    if (mp[mul] && t * (i - 1) - mp[mul] + 1 >= 0) {
      return t * (i - 1) - mp[mul] + 1;
    }
  }
  return -1;
}

ll exBSGS(ll a, ll p, ll b) {
  a %= p, b %= p;
  if (b == 1 || p == 1) {
    return 0;
  }
  ll res = 0, mul = 1;
  for (ll d; (d = __gcd(a, p)) != 1; ) {
    if (b % d) {
      return -1;
    }
    ++ res, b /= d, p /= d;
    mul = (mul * a / d) % p;
    if (mul == b) {
      return res;
    } 
  }
  ll ans = BSGS(a, p, b, mul);
  return (ans == -1? ans : ans + res);
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
  for (ll a, p, b, ans; ; ) {
    cin >> a >> p >> b;
    if (!a) {
      return 0;
    }
    ans = exBSGS(a, p, b);
    if (ans == -1) {
      cout << "No Solution\n";
    } else {
      cout << ans << '\n';
    }
  }
  return 0; 
}

Part 3:例题

P3846 【模板】BSGS / P4195 【模板】扩展 BSGS

模板题,套板子即可。

P4454 [CQOI2018] 破解D-H协议

首先,将题目中 2、3 步的式子写成同余方程形式:

\[\begin{cases} g^a \equiv A \pmod P \\ g^b \equiv B \pmod P\end{cases} \]

这不就是 BSGS 的形式了?于是使用 BSGS 求出 \(a, b\) 之后计算 \(g^{ab} \bmod P\) 即可。

P4884 多少个 1?

感觉这一题难度虚高。

首先,对 \(N\) 个 \(1\) 的数推导通项公式:

\[10^{N - 1} + 10^{N - 2} + \cdots + 10^0 = \frac{9 \cdot 10^{N - 1} + 9 \cdot 10^{N - 2} + \cdots + 9 \cdot 10^0}{9} = \frac{10^{N} - 1}{9} \]

那么方程转化为:

\[\frac{10^{N} - 1}{9} \equiv K \pmod m \]

两边同乘 \(9\),得:

\[10^{N} - 1 \equiv 9K \pmod m \]

移项,得:

\[10^{N} \equiv 9K + 1\pmod m \]

于是就转化为标准 BSGS 形式了。