[学习笔记]杜教筛法的原理

一、引入

通常，在莫比乌斯反演的题目中会推出下面这样的式子：
$$\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor f(i)$$
其中 $f$ 是积性函数。
我们知道，当 $1\le i\le n$ 时 $\lfloor \frac{n}{i}\rfloor$ 的取值只有 $O(\sqrt{n})$ 种。
于是我们可以对下界分块，利用 $f$ 的前缀和求解，从而实现一次 $O(\sqrt{n})$ 的复杂度。
但如果 $n=10^9$ 时，如何求 $f$ 的前缀和呢？
下面就来介绍求积性函数前缀和的算法：杜教筛。

二、前置芝士：狄利克雷 (Dirichlet) 卷积

两个数论函数函数 $f$ 和 $g$ 的狄利克雷卷积定义为：
$$(f\ast g)(n)\sum _{d|n}f(d)g(\frac{n}{d})$$
如果 $f$ 和 $g$ 为积性函数，那么 $f\ast g$ 也是积性函数。

三、算法流程

设我们要求的值：
$$S(n)=\sum _{i=1}^{n}f(i)$$
首先找到一个积性函数 $g$ ，使得
$$h(n)=\sum _{i=1}^n(f\ast g)(i)$$
可以被 $O(1)$ 计算出， $g(n)$ 的值及前缀和也能被 $O(1)$ 计算出。
于是有：
$$\sum _{i=1}^n\sum _{d|i}f(\frac{i}{d})g(d)=h(n)$$
把枚举约数改成枚举倍数：
$$\sum _{d=1}^{n}g(d)\sum _{1\le i\le n,d|i}f(\frac{i}{d})=h(n)$$
满足所有 $1\le i\le n,d|i$ 的 $\frac{i}{d}$ 实际上就是所有满足 $1\le i\le \lfloor \frac{n}{d}\rfloor$ 的 $i$ ：
$$\sum _{i=1}^{n}g(i)\sum _{j=1}^{\lfloor \frac{n}{i}\rfloor }f(j)=h(n)$$
于是：
$$\sum _{i=1}^{n}g(i)S(\lfloor \frac{n}{i}\rfloor )=h(n)$$
由于 $\lfloor \frac{n}{1}\rfloor =n$ ，故：
$$g(1)S(n)=h(n)-\sum _{i=2}^{n}g(i)S(\lfloor \frac{n}{i}\rfloor )$$
对 $\lfloor \frac{n}{i}\rfloor$ 下界分块。使用记忆化搜索就能在 $O(n^{\frac{3}{4}})$ 内求出一个前缀和。
为了提高效率，我们还需要结合线性筛，筛出 $n^{\frac{2}{3}}$ 内的所有 $S$ 值。
这样，我们实现了 $O(n^{\frac{2}{3}})$ 求 $S(n)$ 。

四、复杂度分析

求 $S(n)$ 时由于使用了记忆化，所以会递归到的值 $x$ 一定是 $\lfloor \frac{n}{i}\rfloor$ 的取值（最多 $O(\sqrt{n})$ 种）。
所以复杂度（不结合线性筛的情况下）：
$$\sum _{i=1}^{\sqrt{n}}\sqrt{i}+\sum _{i=1}^{\sqrt{n}}\sqrt{\frac{n}{i}}$$
$$={\int }_0^{\sqrt{n}}(\sqrt{x}+\sqrt{\frac{n}{x}})\text{d}x=O(n^{\frac{3}{4}})$$
如果结合了线性筛，那么能递归到的 $x$ 至少 $n^{\frac{2}{3}}$ ，
复杂度降到：
$$\sum _{i=1}^{n^{\frac{1}{3}}}\sqrt{\frac{n}{i}}={\int }_0^{n^{\frac{1}{3}}}\sqrt{\frac{n}{x}}\text{d}x=O(n^{\frac{2}{3}})$$

五、有趣的栗子

来源： BZOJ 3944
https://www.lydsy.com/JudgeOnline/problem.php?id=3944
（1）求 ${\sum }_{i=1}^n\mu (i)$ 。
我们知道， $\mu \ast 1=e$ （ $e$ 为单位元，即 $[n=1]$ ）
于是：
$$\sum _{i=1}^n\mu (i)=1-\sum _{i=2}^n\sum _{j=1}^{\lfloor \frac{n}{i}\rfloor }\mu (j)$$
对 $\lfloor \frac{n}{i}\rfloor$ 下界分块即可。
（2）求 ${\sum }_{i=1}^n\varphi (i)$ 。
我们知道， $\varphi \ast 1=n$ 。
于是：
$$\sum _{i=1}^n\varphi (i)=\frac{n\times (n+1)}{2}-\sum _{i=2}^n\sum _{j=1}^{\lfloor \frac{n}{i}\rfloor }\varphi (j)$$
同样下界分块。