跳转至

狄利克雷卷积

本文介绍 Dirichlet 卷积和 Dirichlet 生成函数.

Dirichlet 卷积

数论函数 f(n) 和 g(n) 的 Dirichlet 卷积(Dirichlet convolution),记作 f∗g,定义为数论函数

(f∗g)(n)=∑k∣nf(k)g(nk)=∑kℓ=nf(k)g(ℓ).

Dirichlet 卷积是数论函数的重要运算.数论函数的许多性质都是通过这个运算挖掘出来的.

例子
  1. 单位函数 ε 是莫比乌斯函数 μ 和常数函数 1 的 Dirichlet 卷积:

    ε=μ∗1⟺ε(n)=∑d∣nμ(d).
  2. 除数个数函数 τ 是常数函数 1 和它自身的 Dirichlet 卷积:

    τ=1∗1⟺τ(n)=∑d∣n1.
  3. 除数和函数 σ 是恒等函数 id 和常数函数 1 的 Dirichlet 卷积:

    σ=id∗1⟺σ(n)=∑d∣nd.
  4. 欧拉函数 φ 是恒等函数 id 和莫比乌斯函数 μ 的 Dirichlet 卷积:

    φ=id∗μ⟺φ(n)=∑d∣nd⋅μ(nd).

莫比乌斯反演 就是利用 ε=μ∗1 对数论函数恒等式进行变形.

性质

Dirichlet 卷积具有一系列代数性质.

定理

设 f,g,h 都是数论函数.那么,有:

  1. 交换律:f∗g=g∗f.
  2. 结合律:(f∗g)∗h=f∗(g∗h).
  3. 分配律:(f+g)∗h=f∗h+g∗h.
  4. 单位元:f∗ε=ε∗f=f,其中,ε(n)=[n=1] 是卷积单位元,[⋅] 是 Iverson 括号.
  5. 逆元:当且仅当 f(1)≠0 时,存在 g 使得 f∗g=g∗f=ε,且 g 称为 f 的 Dirichlet 逆元(Dirichlet inverse),可以记作 f−1.而且,逆元 g 满足递推公式

    g(n)=ε(n)−∑kℓ=n, k≠1f(k)g(ℓ)f(1).
证明

为验证交换律,直接计算可知

(f∗g)(n)=∑kℓ=nf(k)g(ℓ)=(g∗f)(n).

为验证结合律,直接计算可知

((f∗g)∗h)(n)=∑kℓm=nf(k)g(ℓ)h(m)=(f∗(g∗h))(n).

为验证分配律,直接计算可知

((f+g)∗h)(n)=∑kℓ=n(f(k)+g(k))h(ℓ)=∑kℓ=nf(k)h(ℓ)+∑kℓ=ng(k)h(ℓ)=(f∗h+g∗h)(n).

为验证 ε(n) 是单位元,直接计算可知

(f∗ε)(n)=∑kℓ=nf(k)ε(ℓ)=f(n).

第二个等号是因为 ε(ℓ) 仅在 ℓ=1 即 k=n 时取得非零值.

最后,需要证明 f−1 存在,当且仅当 f(1)≠0.对于任意 f,假设存在 g 使得 f∗g=ε.这意味着

(f∗g)(n)=∑kℓ=nf(k)g(ℓ)=ε(n).

这实际上给出了一系列关于 g(n) 取值的方程组,从中可以直接求出 g(n).特别地,当 n=1 时,等式变为 f(1)g(1)=1,所以 g 存在,至少要求 f(1)≠0.而只要 f(1)≠0,可以直接解出

g(n)=ε(n)−∑kℓ=n, k≠1f(k)g(ℓ)f(1).

它可以用于递归计算 g(n) 的取值.因此,逆元 g 存在,当且仅当 f(1)≠0.

用抽象代数的语言说,这些代数性质说明,全体数论函数在(逐点)加法运算和 Dirichlet 卷积运算下构成 交换环,且它的全体可逆元就是那些在 n=1 处取非零值的函数.这个环称为 Dirichlet 环(Dirichlet ring).

积性函数是一类特殊的数论函数.它对于 Dirichlet 卷积和 Dirichlet 逆都是封闭的.

定理

设 f,g 是积性函数,那么,f∗g 也是积性函数.而且,逆元 f−1 一定存在,它也是积性函数.

证明

对于第一点,设 h=f∗g,直接验证可知,对于 n1⟂n2,都有

h(n1)h(n2)=(∑k1ℓ1=n1f(k1)g(ℓ1))(∑k2ℓ2=n2f(k2)g(ℓ2))=∑k1ℓ1=n1, k2ℓ2=n2f(k1)f(k2)g(ℓ1)g(ℓ2)=∑kℓ=n1n2f(k)g(ℓ)=h(n1n2).

其中,第三个等号利用了 f,g 的积性,其改变求和顺序的逻辑是:当 k 遍历 n1n2 的因数时,k 的素因子可以根据它是 n1 还是 n2 的素因子分为两类,将两类中的素因子(计重复)分别乘起来得到 k1 和 k2,它们将分别遍历 n1 和 n2 的因数;反过来,根据 n1 和 n2 的因数 k1 和 k2,总是可以得到 n1n2 的因数 k=k1k2.

对于第二点,设 g=f−1,考虑应用数学归纳法.首先,g(1)=1/f(1)=1.此时,逆元的递推公式可以写作

g(n)=ε(n)−∑kℓ=n, k≠1f(k)g(ℓ).

所以,对于 n1⟂n2 且 n1n2>1,有

g(n1n2)=−∑kℓ=n1n2, k≠1f(k)g(ℓ)=−∑k1ℓ1=n1, k2ℓ2=n2, k1k2≠1f(k1)f(k2)g(ℓ1)g(ℓ2)=f(1)f(1)g(n1)g(n2)−∑k1ℓ1=n1, k2ℓ2=n2f(k1)f(k2)g(ℓ1)g(ℓ2)=g(n1)g(n2)−(∑k1ℓ1=n1f(k1)g(ℓ1))(∑k2ℓ2=n2f(k2)g(ℓ2))=g(n1)g(n2)−ε(n1)ε(n2)=g(n1)g(n2).

其中,第二个等号用到了归纳假设,即对于 ℓ1ℓ2<n1n2 且 ℓ1⟂ℓ2,条件 g(ℓ1ℓ2)=g(ℓ1)g(ℓ2) 成立.

用抽象代数的语言说,全体积性函数在 Dirichlet 卷积运算下构成 Dirichlet 环乘法群的 子群.

更为特殊的是完全积性函数.

定理

设 α 是完全积性函数,f,g 是数论函数.那么,有:

  1. 分配律:(αf)∗(αg)=α⋅(f∗g).
  2. 逆元:(αf)−1=αf−1,只要 f−1 存在.
  3. 积性函数 f 是完全积性函数,当且仅当 f−1=μf,其中,μ 是 莫比乌斯函数.
证明

对于第一条,直接验证可知

((αf)∗(αg))(n)=∑kℓ=n(αf)(k)(αg)(ℓ)=∑kℓ=nα(k)f(k)α(ℓ)g(ℓ)=∑kℓ=nα(n)f(k)g(ℓ)=α(n)(f∗g)(n).

其中,第三个等号用到了完全积性函数的性质:α(n)=α(k)α(ℓ) 对所有 n=kℓ 都成立.

对于第二条,利用第一条就有

(αf)∗(αf−1)=α(f∗f−1)=αε=ε.

其中,最后一个等号只利用了 α(1)=1.由逆元定义,(αf)−1=αf−1.

对于第三条,利用第二条和 1−1=μ 可知,如果 f 是完全积性函数,那么

f−1=(1f)−1=1−1⋅f=μf.

其中,1 是常数函数.反过来,如果 f 是积性函数且 f−1=μf,那么只需要证明对于所有素数 p 和 e∈N+,都有 f(pe)=f(p)e 成立,就能证明 f 是完全积性函数.为此,对 e∈N+ 应用数学归纳法.归纳起点 e=1 处命题显然成立.对于任意 e>1,应用逆元递推公式,都有

f−1(pe)=−∑i=1ef(pi)f−1(pe−i)=−∑i=1ef(pi)μ(pe−i)f(pe−i)=−f(pe)f(1)μ(1)−f(pe−1)μ(p)f(p)=−f(pe)+f(p)e.

其中,最后一个等号用到了归纳假设 f(pe−1)=f(p)e−1.应用 f−1=μf,就得到

f−1(pe)=μ(pe)f(pe)=0.

代入前式,就得到

f(pe)=f(p)e.

所以,归纳步骤成立.原命题得证.

用抽象代数的语言说,如果 α 是完全积性函数,映射 f↦αf 是 Dirichlet 环的 自同态.

Dirichlet 生成函数

与 Dirichlet 卷积紧密相关的是 Dirichlet 生成函数.

数论函数 f(n)——也就是数列 {f(n)}——对应的 Dirichlet 生成函数(Dirichlet series generating function,DGF)定义为形式 Dirichlet 级数(formal Dirichlet series):

F(s)=∑n=1∞f(n)ns.

级数中的 s 是形式变元.常见的 Dirichlet 生成函数中,s 往往可以看作是复变量,进而讨论 Dirichlet 级数的解析性质,但这超出了算法竞赛的范围.

Dirichlet 生成函数的乘积对应着相应的数论函数的 Dirichlet 卷积:

定理

对于数论函数 f,g 及其 Dirichlet 生成函数 F,G,它们的 Dirichlet 卷积 f∗g 的生成函数等于 F⋅G.

证明

直接验证:

F(s)G(s)=(∑k=1∞f(k)ks)(∑ℓ=1∞g(ℓ)ℓs)=∑k=1∞∑ℓ=1∞f(k)g(ℓ)(kℓ)s=∑n=1∞∑kℓ=nf(k)g(ℓ)ns=∑n=1∞(f∗g)(n)ns.

利用 Dirichlet 卷积和 Dirichlet 生成函数乘积之间的对应关系,可以从 Dirichlet 生成函数的角度理解 Dirichlet 卷积的性质.由于形式 Dirichlet 级数的乘法运算满足交换律、结合律、对加法的分配律,数论函数的 Dirichlet 卷积运算满足同样的代数性质.

Euler 乘积

积性函数的特殊性同样反映在 Dirichlet 生成函数上.由于整数有 唯一分解定理,积性函数 f(n) 的生成函数 F(s) 可写成如下形式:

F(s)=∑n=1∞f(n)ns=∑n=1∞∏p∈Pf(pνp(n))pνp(n)s=∏p∈P∑e=0∞f(pe)pes=∏p∈P(1+f(p)ps+f(p2)p2s+f(p3)p3s+⋯).

其中,νp(n) 表示 n 的素因数分解中 p 的幂次.这意味着,F(s) 可以分解为若干 Fp(s)=∑e=0∞f(pe)p−es 的乘积,且每个 Fp(s) 对应的数论函数都只在 p 的幂次处可能取非零值.这一无穷乘积也称为 Euler 乘积(Euler product).如果 F(s) 和 G(s) 都能分解成类似的形式,那么它们的乘积同样如此;将这一观察对应到数论函数上,就是积性函数的 Dirichlet 卷积仍然是积性函数.

进一步地,如果 f(n) 还是完全积性函数,那么 f(pe)=f(p)e,上式可以继续简化:

F(s)=∏p∈P∑e=0∞f(p)epes=∏p∈P(1−f(p)ps)−1.

与积性函数不同,完全积性函数的 Dirichlet 生成函数的形式在乘法运算下并不具有封闭性,因此,完全积性函数的 Dirichlet 卷积和 Dirichlet 逆都未必是完全积性函数,但一定是积性函数.

例子
  1. 单位函数 ε(n) 是完全积性函数.它的 Dirichlet 生成函数是关于不定元 s 的常值函数

    E(s)=∑n=1∞ε(n)ns=1.
  2. 常数函数 1(n) 是完全积性函数.它的 Dirichlet 生成函数是 Riemann ζ 函数

    I(s)=∑n=1∞1ns=∏p∈P11−p−s=ζ(s).
  3. 莫比乌斯函数 μ(n) 是常数函数的 Dirichlet 逆.它的 Dirichlet 生成函数是 ζ(s) 的倒数:

    M(s)=∑n=1∞μ(n)ns=∏p∈P(1−p−s)=1ζ(s).
  4. 幂函数 idk⁡(n)=nk 是完全积性函数.特别地,当 k=0 时,它就是常数函数;当 k=1 时,它就是恒等函数.它的 Dirichlet 生成函数是

    Ik(s)=∑n=1∞nkns=∏p∈P11−pk−s=ζ(s−k).
  5. 欧拉函数 φ(n) 是积性函数.它的 Dirichlet 生成函数是

    Φ(s)=∏p∈P(1+p−1ps+p(p−1)p2s+p2(p−1)p3s+⋯)=∏p∈P(11−p1−s−1ps11−p1−s)=∏p∈P1−p−s1−p1−s=ζ(s−1)ζ(s).

    结合幂函数的 Dirichlet 生成函数表达式,就得到 id=φ∗1.

  6. 约数函数 σk(n)=∑d∣ndk 是积性函数.它的 Dirichlet 生成函数是

    Σk(s)=∏p∈P(1+1+pkps+1+pk+p2kp2s+1+pk+p2k+p3kp3s+⋯)=∏p∈P11−pk((1−pk)+1−p2kps+1−p3kp2s+1−p4kp3s+⋯)=∏p∈P11−pk(11−p−s−pk1−pk−s)=∏p∈P1(1−p−s)(1−pk−s)=ζ(s−k)ζ(s).

    结合幂函数的 Dirichlet 生成函数表达式,就得到 σk=idk∗1.这正是 σk 的定义式.

  7. 无平方因子数的指示函数 u(n)=|μ(n)|=μ2(n) 是积性函数.它的 Dirichlet 生成函数是

    U(s)=∏p∈P(1+p−s)=∏p∈P1−p−2s1−p−s=ζ(s)ζ(2s).

Bell 级数

Euler 乘积说明,积性函数的 Dirichlet 生成函数可以分解为局部因子 Fp(s) 的乘积.如果只关心单个素数处的信息,可以将 p−s 整体看作一个形式变元,这就得到了 Bell 级数.

对于积性函数 f 和素数 p,定义 f 关于 p 的 Bell 级数(Bell series)为形式幂级数

fp(x)=∑e=0∞f(pe)xe=1+f(p)x+f(p2)x2+⋯.

对比 Euler 乘积的表达式可知,fp(x) 就是将 Fp(s) 中的 p−s 替换为 x 的结果,即 Fp(s)=fp(p−s),从而

F(s)=∏p∈Pfp(p−s).

引入这一记号后可以看出,Dirichlet 卷积在局部化之后就是普通的幂级数乘法.

将 Dirichlet 生成函数的结果转译到 Bell 级数上,就得到如下性质:

  1. 对于积性函数 f,g 及任意素数 p,有 (f∗g)p(x)=fp(x)gp(x).
  2. 积性函数 f 是完全积性函数,当且仅当对每个素数 p 都有 fp(x)=11−f(p)x.
  3. 积性函数的 Dirichlet 逆总是存在,且 (f−1)p(x)=1fp(x).

因为积性函数由它在素数幂处的取值唯一确定,所以积性函数与它的全体 Bell 级数 {fp}p∈P 是一一对应的.于是,关于积性函数的 Dirichlet 卷积的许多问题,都可以逐个素数地转化为形式幂级数的问题来讨论.相比直接使用 Dirichlet 生成函数,其形式上要简洁一些.例如,第三条性质说明,求 Dirichlet 逆只需逐个素数地求形式幂级数的逆;在实际计算中,因为只需要 pe≤n 的部分,每个素数至多截断到 x⌊logp⁡n⌋ 项.

例子

前文例子中的积性函数,对应的 Bell 级数如下:

fε1μidkφσk|μ|=μ2
fp(x)111−x1−x11−pkx1−x1−px1(1−x)(1−pkx)1+x

利用上表,常见的 Dirichlet 卷积关系都可以直接验证.例如 μ∗1=ε 对应

(1−x)⋅11−x=1,

而 φ∗1=id 对应

1−x1−px⋅11−x=11−px.

应用

Dirichlet 生成函数可以用于将数论函数表示为 Dirichlet 卷积.

例如在杜教筛的过程中,要计算数论函数 f 的前缀和,需要找到另一个数论函数 g 使得 f∗g 和 g 都可以快速求前缀和.可以利用 Dirichlet 生成函数推导这一过程.

以杜教筛一节的例题 Luogu P3768 简单的数学题 为例,需要对 f(n)=n2φ(n) 构造满足上述条件的数论函数 g(n).由于 f 是积性函数,它的 Dirichlet 生成函数为

F(s)=∏p∈P(1+∑k=1∞p3k−1(p−1)pks)=∏p∈P1−p2−s1−p3−s=ζ(s−3)ζ(s−2).

对比幂函数的 Dirichlet 生成函数可知,只要取 g=id2,就有 f∗g=id3.两者都是可以快速计算前缀和的.

由于常见问题中的数论函数都是积性函数,实践中利用 Bell 级数往往更加方便.例如,刚刚的例子中 f(n) 的 Bell 级数可以直接由 f(pe)=p3e−1(p−1) 推得为

fp(x)=1+p−1p∑e=1∞(p3x)e=1+(p−1)p2x1−p3x=1−p2x1−p3x.

对比幂函数的 Bell 级数立即得到 f∗id2=id3.

Dirichlet 卷积的计算

本节讨论 Dirichlet 卷积的计算问题,即给定序列 {f(k)}k=1n 和 {g(k)}k=1n,求解 Dirichlet 卷积 h=f∗g 的前若干项 {h(k)}k=1n 的问题.根据涉及到的函数性质,算法的复杂度也略有不同.

一般情形

如果 f,g,h 都没有特殊性质,那么 Dirichlet 卷积的计算只能利用其定义:

h(n)=∑kℓ=nf(k)g(ℓ).

枚举 k 和 ℓ,将贡献 f(k)g(ℓ) 累加到 h(kℓ) 上即可.枚举复杂度为

O(∑k=1nnk)=O(nlog⁡n).

参考实现如下:

参考实现
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
// Compute the Dirichlet convolution h = f * g.
auto dirichlet_convolute(const std::vector<int>& f, const std::vector<int>& g) {
  int n = f.size() - 1;
  std::vector<int> h(n + 1);
  for (int k = 1; k <= n; ++k) {
    for (int d = 1; k * d <= n; ++d) {
      h[k * d] += f[k] * g[d];
    }
  }
  return h;
}

与积性函数卷积的情形

如果 g 是积性函数,那么可以利用 Euler 乘积加速 Dirichlet 卷积的计算.计算 h 相当于计算它的 Dirichlet 生成函数 H 中各项的系数.由于

H(s)=F(s)G(s)=F(s)∏p∈PGp(s).

其中,Gp(s) 是 G(s) 的 Euler 乘积分解中的因式,它只包含 p 的幂次处的系数:

Gp(s)=∑pk≤ng(pk)pks=1+g(p)ps+g(p2)p2s+⋯.

那么,从 F(s) 开始,遍历所有不超过 n 的素数 p,将 Gp(s) 逐一乘上去,同样可以得到最终结果 H(s).将 Gp(s) 乘上去时,直接应用一般情形中的暴力枚举算法即可.总枚举次数

∑p∈P, p≤n∑k=1∞⌊npk⌋≤∑p∈P, p≤nnp−1≤∑p∈P, p≤n2np∈O(nlog⁡log⁡n).

最后一步复杂度的估计与 Eratosthenes 筛法 复杂度的证明一致.所以,本算法的时间复杂度为 O(nlog⁡log⁡n).

参考实现如下:

参考实现
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
// Compute the Dirichlet convolution h = f * g.
// Assume that g is multiplicative.
auto dirichlet_convolute(const std::vector<int>& f, const std::vector<int>& g) {
  int n = f.size() - 1;
  std::vector<int> h(f);
  std::vector<bool> vis(n + 1);
  for (int i = 2; i <= n; ++i) {
    if (vis[i]) continue;
    // Reverse the order for in-place computation.
    for (int k = n / i * i; k; k -= i) {
      vis[k] = true;
      int d = k;
      while (true) {
        d /= i;
        h[k] += h[d] * g[k / d];
        if (d % i) break;
      }
    }
  }
  return h;
}

特别地,当积性函数 g 是完全积性函数或其 Dirichlet 逆时,例如当 g=1 或 g=μ 时,算法可以进一步简化.此时,Dirichlet 卷积 h=f∗g 的计算可以采用常数更小的 Dirichlet 前缀和/差分 算法,但是算法时间复杂度仍为 O(nlog⁡log⁡n).

结果为积性函数的情形

最后,考虑 h 是积性函数的情形.特别地,当 f,g 都是积性函数时,h=f∗g 就是积性函数.要计算 h,只需要确定它在素数幂处的取值,就可以通过 线性筛 在 O(n) 时间内计算.而对于素数幂 pe 处的取值 h(pe) 直接暴力计算即可:

h(pe)=∑i=0ef(pi)g(pe−i).

这些暴力计算需要的枚举次数为

∑p∈P, p≤n∑e=1⌊logp⁡n⌋(e+1)≤∑p∈P, p≤n2⌊logp⁡n⌋2+∑p∈P, n<p≤n2≤2n(log2⁡n)2+2n∈O(n).

因此,这一算法的总时间复杂度为 O(n).

参考实现如下:

参考实现
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
// Compute the Dirichlet convolution h = f * g.
// Assume that h is multiplicative.
auto dirichlet_convolute(const std::vector<int>& f, const std::vector<int>& g) {
  int n = f.size() - 1;
  std::vector<int> h(n + 1), primes, rem(n + 1), lpf(n + 1);
  std::vector<bool> vis(n + 1);
  h[1] = 1;
  for (int x = 2; x <= n; ++x) {
    if (!vis[x]) {
      primes.push_back(x);
      rem[x] = 1;
      lpf[x] = x;
    }
    for (int p : primes) {
      if (x * p > n) break;
      vis[x * p] = true;
      rem[x * p] = x % p ? x : rem[x];
      lpf[x * p] = p;
      if (x % p == 0) break;
    }
    if (rem[x] == 1) {  // prime powers.
      for (int k = x; k; k /= lpf[x]) {
        h[x] += f[k] * g[x / k];
      }
    } else {  // other cases.
      h[x] = h[rem[x]] * h[x / rem[x]];
    }
  }
  return h;
}

参考资料与注释