浅谈一类积性函数的前缀和

最新推荐文章于 2022-06-27 01:56:52 发布

oorzhang

最新推荐文章于 2022-06-27 01:56:52 发布

阅读量1.4k

点赞数 1

分类专栏：数学数论

数学数论专栏收录该内容

31 篇文章 0 订阅

订阅专栏

前置技能

积性函数的定义

若 f(n) 的定义域为正整数域，值域为复数，即 f:Z+→C ，则称 f(n) 为数论函数。
若 f(n) 为数论函数，且 f(1)=1 ，对于互质的正整数 p,q 有 f(p⋅q)=f(p)⋅f(q) ，则称其为积性函数。
若 f(n) 为积性函数，且对于任意正整数 p,q 都有 f(p⋅q)=f(p)⋅f(q) ，则称其为完全积性函数。

积性函数的性质与例子

常见的积性函数有
- 除数函数 σk(n)=∑d|ndk ，表示 n 的约数的 k 次幂和，注意 σk(n) 与 σk(n) 是不同的。
- 约数个数函数 τ(n)=σ0(n)=∑d|n1 ，表示 n 的约数个数，一般也写为 d(n) 。
- 约数和函数 σ(n)=σ1(n)=∑d|nd ，表示 n 的约数之和。
- 欧拉函数 φ(n)=∑ni=1[(n,i)=1]⋅1 ，表示不大于 n 且与 n 互质的正整数个数，另外 ∑ni=1[(n,i)=1]⋅i=n⋅φ(n)+[n=1]2 ，且对于正整数 n>2 来说 φ(n) 是偶数。
- 莫比乌斯函数 μ(n) ，在狄利克雷卷积的乘法中与恒等函数互为逆元， μ(1)=1 ，对于无平方因子数 n=∏ti=1pi 有 μ(n)=(−1)t ，对于有平方因子数 n 有 μ(n)=0 。
- 元函数 e(n)=[n=1] ，狄利克雷卷积的乘法单位元，完全积性。
- 恒等函数 I(n)=1 ，完全积性。
- 单位函数 id(n)=n ，完全积性。
- 幂函数 idk(n)=nk ，完全积性。
关于莫比乌斯函数和欧拉函数有两个经典的公式
- [n=1]=∑d|nμ(d) ，将 μ(d) 看作是容斥的系数即可证明。
- n=∑d|nφ(d) ，将 in(1≤i≤n) 化为最简分数统计个数即可证明。
若 f(n) 为积性函数，则对于正整数 n=∏ti=1pkii 有 f(n)=∏ti=1f(pkii) ；若 f(n) 为完全积性函数，则对于正整数 n=∏ti=1pkii 有 f(n)=∏ti=1f(pi)ki 。

狄利克雷卷积与莫比乌斯反演

数论函数 f 和 g 狄利克雷卷积定义为 (f∗g)(n)=∑d|nf(d)⋅g(nd) ，狄利克雷卷积满足交换律、结合律，对加法满足分配律，存在单位元函数 e(n)=[n=1] 使得 f∗e=f=e∗f ，若 f 和 g 为积性函数则 f∗g 也为积性函数。
狄利克雷卷积的一个常用技巧是对于积性函数 f 与恒等函数 I 的卷积的处理，例如 n=∏ti=1pkii,g(n)=∑d|nf(d) ，则有 g(n)=∏ti=1∑kij=0f(pji) 。
莫比乌斯反演也是对于 g(n)=∑d|nf(d) 的讨论，但是不要求 f 是积性函数，适用于已知 g(n) 求 f(n) 的情况，由于 I∗μ=e ，则 g∗μ=f∗I∗μ=f∗e=f ，即 f(n)=∑d|ng(d)⋅μ(nd) ，类似地有 g(n)=∑n|df(d)⇒f(n)=∑n|dg(d)⋅μ(dn) ，二项式反演也是类似的技巧。有一个例子可以看出欧拉函数和莫比乌斯函数之间的关系，由于 ∑d|nφ(d)=id(n) ，所以 φ(n)=∑d|nμ(d)nd ，也即 φ(n)n=∑d|nμ(d)d 。

正文：黑科技

这种黑科技大概起源于Project Euler这个网站，由xudyh引入中国的OI、ACM界，目前出现了一些OI模拟题、OJ月赛题、ACM赛题是需要这种技巧在低于线性时间的复杂度下解决一类积性函数的前缀和问题。

首先看一个简单的例子，求前 n 个正整数的约数之和，即 ∑ni=1σ(i) ，其中 n≤1012 。
显然不能直接做了，但是我们可以推导一番：

\sum i = 1 n σ (i) = \sum i = 1 n \sum j = 1 n [j | i] \cdot j = \sum i = 1 n i \cdot \sum j = 1 n [i | j] = \sum i = 1 n i \cdot ⌊ n i ⌋

当

i≤n−−√ 时，

⌊ni⌋ 显然只有

O(n−−√) 个取值；当

i>n−−√ 时，

⌊ni⌋<n−−√ 显然也只有

O(n−−√) 个取值；对于固定的

⌊ni⌋ ，

i 的取值是一段连续的区间，当

⌊ni⌋>1 时这段区间是

[⌊n⌊ni⌋−1⌋+1,⌊n⌊ni⌋⌋] ，因此可以

O(n−−√) 计算所求。
同样地，求前

n 个正整数的约数个数之和也可以这样计算，留给读者练习。
另外需要说明的是，

∑ni=1⌊ni⌋⋅i=∑ni=1⌊ni⌋⋅(⌊ni⌋+1)2 ，这也是一种常见的表示形式。

现在我们来加大一点难度，求前 n 个正整数的欧拉函数之和，即 ∑ni=1φ(i) ，其中 n≤1011 。
目前本文提到的有关欧拉函数的公式只有几个，是否能用它们来帮助化简呢？答案是肯定的，接下来我们就利用 ∑d|nφ(d)=n 来化简这个式子。
这个公式也可以看成是 φ(n)=n−∑d|n,d<nφ(d) ，设 ϕ(n)=∑ni=1φ(i) ，则有

ϕ (n) = \sum i = 1 n φ (i) = \sum i = 1 n i - \sum d | i, d < i φ (d) = n \cdot ( n + 1 ) 2 - \sum i = 2 n \sum d | i, d < i φ (d) = n \cdot ( n + 1 ) 2 - \sum i d = 2 n \sum d = 1 ⎢ ⎣ ⎢ ⎢ n i d ⎥ ⎦ ⎥ ⎥ φ (d) = n \cdot ( n + 1 ) 2 - \sum i = 2 n \sum d = 1 ⌊ n i ⌋ φ (d) = n \cdot ( n + 1 ) 2 - \sum i = 2 n ϕ (⌊ n i ⌋)

那么只要计算出

O(n−−√) 个

ϕ(⌊ni⌋) 的值，就可以计算出

ϕ(n) ，这样的复杂度又如何呢？
假设计算出

ϕ(n) 的复杂度为

T(n) ，则有

T(n)=O(n−−√)+∑n√i=1T(i)+T(ni) ，只展开一层就可以了，更深层的复杂度是高阶小量，所以有

T(n)=∑n√i=1O(i√)+O(ni−−√)=O(n34) 。
由于

ϕ(n) 是一个积性函数的前缀和，所以筛法也可以预处理一部分，假设预处理了前

k 个正整数的

ϕ(n) ，且

k≥n−−√ ，则复杂度变为

T(n)=∑nki=1ni−−√=O(nk√) ，当

k=O(n23) 时可以取到较好的复杂度

T(n)=O(n23) 。
之前利用

φ(n)=n−∑d|n,d<nφ(d) 的地方是怎么想到的呢？不妨来看一下这个

n \cdot ( n + 1 ) 2 = \sum i = 1 n i = \sum i = 1 n \sum d [d | i] \cdot φ (d) = \sum i d = 1 n \sum d = 1 ⎢ ⎣ ⎢ ⎢ n i d ⎥ ⎦ ⎥ ⎥ φ (d) = \sum i = 1 n ϕ (⌊ n i ⌋)

如果能通过狄利克雷卷积构造一个更好计算前缀和的函数，且用于卷积的另一个函数也易计算，则可以简化计算过程。例如上题就是利用了

φ∗I=id 的性质，但一定注意，不是所有的这一类题都只用配个恒等函数

I 就可以轻松完事的，有时需要更细致的观察。

定义梅滕斯函数 M(n)=∑ni=1μ(i) ，给定正整数 n ，计算 M(n) ，其中 n≤1011 。
有了欧拉函数的经验，这次似乎就轻车熟路了吧，使用 [n=1]=∑d|nμ(d) 来试试？

1 = \sum i = 1 n [i = 1] = \sum i = 1 n \sum d | i μ (d) = \sum i = 1 n \sum d = 1 ⌊ n i ⌋ μ (d) = \sum i = 1 n M (⌊ n i ⌋)

因此

M(n)=1−∑ni=2M(⌊ni⌋) ，问题可在

O(n23) 时间复杂度下解决。

看了上面的例子，是不是认为这种做法很naive，很好学啊，再来看一个题吧！
令 A(n)=∑ni=1i(n,i),F(n)=∑ni=1A(i) ，求 F(n) 模 (109+7) 的值，其中 n≤109 。
先做一番化简，变成积性函数前缀和的样子：

A (n) = \sum i = 1 n i ( n , i ) = \sum i = 1 n \sum d | n [(n, i) = d] \cdot i d = \sum d | n \sum i d = 1 n d [(n d, i d) = 1] \cdot i d = 1 2 (1 + \sum d | n d \cdot φ (d))

设

G(n)=2⋅F(n)−n ，则

G (n) = 2 \cdot F (n) - n = \sum i = 1 n \sum d | i d \cdot φ (d) = \sum i = 1 n \sum d = 1 ⌊ n i ⌋ d \cdot φ (d)

因此要求的是

ϕ′(n)=∑ni=1i⋅φ(i) 。
而对于

n=∏ti=1pkii ，有

\sum d | n d \cdot φ (d) = \prod i = 1 t \sum j = 0 k i p j i \cdot φ (p j i) = \prod i = 1 t p 2 k i + 1 i + 1 p i + 1

这并不是什么好算前缀和的函数。
但是不难发现

(id⋅φ)∗id=id2 ，即

\sum d | n d \cdot φ (d) \cdot n d = n \cdot \sum d | n φ (d) = n 2

，这是一个很好计算前缀和的函数，于是有

n \cdot ( n + 1 ) \cdot ( 2 n + 1 ) 6 = \sum i = 1 n i 2 = \sum i = 1 n \sum d | i d \cdot φ (d) \cdot i d = \sum i d = 1 n i d \cdot \sum d = 1 ⎢ ⎣ ⎢ ⎢ n i d ⎥ ⎦ ⎥ ⎥ d \cdot φ (d) = \sum i = 1 n i \cdot ϕ' (⌊ n i ⌋)

因此

ϕ′(n)=n⋅(n+1)⋅(2n+1)6−∑ni=2i⋅ϕ′(⌊ni⌋) ，原问题可在预处理前

O(n23) 个值的基础上，在

O(n23logn) 的时间复杂度下解决。

但是注意到这种方法的常数与复杂度都可能较高，有时候可能再进行一些推导可以得到一个不使用正文方法的做法，例如ZOJ 3881 - From the ABC conjecture，网上存在一个解法是类似本文的方法，但是可以这样推导之后得到更简单的一个做法。

需要使用此种方法的题目一般数据规模较大，例如 n≤109,n≤1011,n≤1012 ，但是并不是都要使用此类方法，有时候可能存在其他 O(n−−√),O(n23) 的做法，例如51Nod 1222 - 最小公倍数计数，会利用正文复杂度分析的方法即可，再例如ZOJ 5340 - The Sum of Unitary Totient，笔者不是很懂这题是否有其他做法，网上的一个解法是分段压缩打表，总之具体问题具体分析。