关于一类积性函数求和的优化——杜教筛

前置技能点

这是一些前置技能点，扫一眼目录，如果你都会就跳过吧。

有关积性函数的一些定义

1. 如果一个函数 $f(x)$ 的定义域为正整数域 ${\mathbb{Z}}^{+}$，值域为复数域 $\mathbb{C}$（即 $f:{\mathbb{Z}}^{+}\to \mathbb{C}$），那么称 $f(x)$ 为数论函数。
2. 若 $f(x)$ 为数论函数，并具有以下性质：$f(1)=1$，且$\forall (p,q)=1$，有$f(p\cdot q)=f(p)\cdot f(q)$。那么我们称 $f(x)$ 为积性函数。
3. 若 $f(x)$ 为积性函数，且$\forall p,q$，有$f(p\cdot q)=f(p)\cdot f(q)$。我们称 $f(x)$ 为完全积性函数。

常见的一些积性函数

1. 除数函数以及其两个特例

除数函数 ${\sigma }_k(n)={\sum }_{d|n}{d}^k$，表示 $n$ 的约数的 $k$ 次幂的和。

约数个数函数 $\tau (n)={\sigma }_0(n)={\sum }_{d|n}1$，表示一个数的约数个数。

约数和函数 $\sigma (n)={\sigma }_1(n)={\sum }_{d|n}d$，表示一个数的所有约数之和。

2. 欧拉函数

$\phi (n)={\sum }_{i=1}^n\left[\left(n,i\right)=1\right]$ 表示不大于 $n$ 且与 $n$ 互质的正整数个数 。 ¹

性质1： ${\sum }_{n|d}\phi (d)=n$。 ²

性质2： ${\sum }_{i=1}^n[(n,i)=1]\cdot i=\frac{n\cdot \phi (n)+[n=1]}{2}$。

性质3： $\forall n\in {\mathbb{N}}^{+}$，若 $n>2$，则有 $2|\phi (n)$。

3. 莫比乌斯函数

$$\mu (n)=\{\begin{array}{ll}1,& \mathrm{i}\mathrm{f}n=1\\ (-1{)}^k,& \mathrm{i}\mathrm{f}n=p_1{p}_2\cdots p_k(p_i\mathrm{i}\mathrm{s}\mathrm{a}\mathrm{p}\mathrm{r}\mathrm{i}\mathrm{m}\mathrm{e})\\ 0,& \mathrm{o}\mathrm{t}\mathrm{h}\mathrm{e}\mathrm{r}\mathrm{w}\mathrm{i}\mathrm{s}\mathrm{e}.\end{array}$$

性质1：莫比乌斯函数在狄利克雷卷积的乘法中与下文的恒等函数互为逆元。

性质2： $[n=1]={\sum }_{d|n}\mu (d)$ ³

4.其它的常见积性函数

元函数 $e(n)=[n=1]$，狄利克雷卷积的乘法单位元，完全积性。

恒等函数 $I(n)=1$，完全积性。

单位函数 $id(n)=n$ ，完全积性。

幂函数 $i{d}^k(n)=n^k$，完全积性。

5.一个常用性质

若一个函数 $f(n)$ 为积性函数，则对于一个正整数 $n={\prod }_{i=1}^t{p}_i^{k_i}$，有 $f(n)={\prod }_{i=1}^{t}f(p_i^{k_i})$。

若一个函数 $f(n)$ 为完全积性函数，则对于一个正整数 $n={\prod }_{i=1}^t{p}_i^{k_i}$，有 $f(n)={\prod }_{i=1}^t{f}^{k_i}(p_i)$。

狄利克雷卷积

1. 两个数论函数 $f$ 和 $g$ 的狄利克雷卷积定义为 $(f\ast g)(n)={\sum }_{d|n}f(d)\cdot g(\frac{n}{d})$。狄利克雷卷积满足交换律、结合律，对加法满足分配律，存在单位元函数 $e(n)=[n=1]$ 使得 $f\ast e=f=e\ast f$，若 $f$ 和 $g$ 为积性函数则 $f\ast g$ 也为积性函数。
2. 常见的考虑利用狄利克雷卷积的方法是对于单位元函数或者恒等函数的处理。举个例子，若是有 $n={\prod }_{i=1}^t{p}_i^{k_i},g(n)={\sum }_{d|n}f(d)$，则 $g(n)={\prod }_{i=1}^t{\sum }_{j=0}^{k_i}f(p_i^j)$。

正文

先举几个例子吧。

例1：欧拉函数的前缀和

求前 $n$ 个正整数的欧拉函数之和，即 $\varphi (n)={\sum }_{i=1}^n\phi (i)$。
前文告诉我们了一个公式，是 ${\sum }_{d|n}\phi (d)=n$，我们利用这个式子进行化简：
$$\begin{gathered} \varphi (n)=\sum _{i=1}^n\phi (i)=\sum _{i=1}^{n}i-\sum _{d|i,d<i}\phi (d) \\ =\frac{n\cdot (n+1)}{2}-\sum _{i=2}^n\sum _{d|i,d<i}\phi (d) \\ =\frac{n\cdot (n+1)}{2}-\sum _{\frac{i}{d}=2}^n\sum _{d=1}^{\lfloor \frac{n}{\frac{i}{d}}\rfloor }\phi (d) \\ =\frac{n\cdot (n+1)}{2}-\sum _{i=2}^n\sum _{d=1}^{\lfloor \frac{n}{i}\rfloor }\phi (d) \\ =\frac{n\cdot (n+1)}{2}-\sum _{i=2}^n\varphi (\lfloor \frac{n}{i}\rfloor ) \end{gathered}$$
那么只要计算出 $O(\sqrt{n})$ 个 $\varphi (\lfloor \frac{n}{i}\rfloor )$ 的值就可以计算出 $\varphi (n)$ 的值了。接下来让我们研究一下这样计算的时间复杂度。
我们假设计算一个 $\varphi (n)$ 的时间复杂度为 $T(n)$，那么 $T(n)=O(\sqrt{n})+{\sum }_{i=1}^{\sqrt{n}}T(i)+T(\frac{n}{i})$，这里只展开一层就可以了，更深层的复杂度是高阶小量我们不进行考虑了，于是有 $T(n)={\sum }_{i=1}^{\sqrt{n}}O(\sqrt{i})+O(\sqrt{\frac{n}{i}})=O(n^{\frac{3}{4}})$。
再优化一下，我们知道 $\phi (n)$ 是一个积性函数，可以运用线性筛法进行预处理。于是我们预处理前 $k$ 个 $\phi (n)$ 并前缀和计算 $\varphi (n)$（这里我们让 $k\ge \sqrt{(}n)$），则有 $T(n)={\sum }_{i=1}^{\frac{n}{k}}\sqrt{\frac{n}{i}}=O(\frac{n}{\sqrt{k}})$。于是可以知道当 $k$ 取到 $n^{\frac{2}{3}}$ 的时候复杂度最优，为 $O(n^{\frac{2}{3}})$。
那么问题来了，我们利用的 ${\sum }_{d|n}\phi (d)=n$ 到底本质上是什么原理呢？让我们改变一下形式看看。

$$\begin{gathered} \sum _{i=1}^{n}i=\sum _{i=1}^n\sum _d[d|i]\cdot \phi (d) \\ =\sum _{\frac{i}{d}=1}^n\sum _{d=1}^{\lfloor \frac{n}{\frac{i}{d}}\rfloor }\phi (d)=\sum _{i=1}^n\varphi (\lfloor \frac{n}{i}\rfloor ) \end{gathered}$$
这是不是就是运用到了狄利克雷卷积，把欧拉函数和恒等函数卷了起来，得到了 $\phi \ast I=id$。

因此本质上，我们要做的事情便是通过狄利克雷卷积构造一个更好计算前缀和的函数，且用于卷积的另一个函数也易计算，则可以简化计算过程。

形式化的推导

我们考虑当前有一个积性函数 $f(n)$，要求计算 $S(n)={\sum }_{i=1}^{n}f(i)$。
我们取积性函数 $g(i)$，计算 $f$ 和 $g$ 的狄利克雷卷积，有 $(f\ast g)(n)={\sum }_{d|n}g(d)f(\frac{n}{d})$
于是有
$$\sum _{i=1}^n(f\ast g)(i)=\sum _{i=1}^n\sum _{d|i}f(d)\cdot g(\frac{n}{d})$$
我们改变枚举顺序，将原本的枚举 $i$ 以及 $i$ 的约数改为枚举 $d$ 并枚举 $d$ 的倍数。
则有
$$\begin{gathered} 原式=\sum _{d=1}^n\sum _{i=1}^{\frac{n}{d}}f(i)g(d) \\ =\sum _{d=1}^{n}g(d)\sum _{i=1}^{\frac{n}{d}}f(i) \\ =\sum _{d=1}^{n}g(d)S(\frac{n}{d}) \end{gathered}$$
于是有
$$g(1)S(n)=\sum _{i=1}^n(f\ast g)(i)-\sum _{d=1}^{n}g(d)S(\frac{n}{d})$$
其中 ${\sum }_{d=1}^{n}g(d)S(\frac{n}{d})$ 可以通过数论分块来以较优的复杂度进行计算。
形式化地，我们只需要构造一个便于计算的函数 $g(n)$，将其与要求前缀和的原函数 $f(n)$ 进行狄利克雷卷积，若是卷积之后的函数也便于计算，我们就可以以较优的复杂度解决问题。

例2：第n个含大于1平方因子的数

求第 $n$ 个含大于 $1$ 平方因子的数。（$n\le 10^{18}$ )

题解

关于平方因子，我们考虑利用莫比乌斯函数的性质（其取0值的时候必定含有大于 $1$ 的平方因子。
因此我们记 $S(n)={\sum }_{i=1}^n{\mu }^2(i)$，于是我们要取的 $ans$ 就是 $ans-s_{ans}=n$。
接下来我们考虑二分，这是显而易见的，重点在于如何高效计算 $S(n)$。
我们知道有一个关于莫比乌斯函数平方的性质，即 ${\mu }_n^2={\sum }_{d^2|n}{\mu }_d$，那么就有
$$s_n=\sum _{i=1}^n{\mu }_n^2=\sum _{i=1}^n\sum _{d^2|n}{\mu }_d=\sum _{d=1}^{\sqrt{n}}{\mu }_d\lfloor \frac{n}{d^2}\rfloor$$
考虑运用数论分块的思想，把 $S(n)$ 分成 $S_1(n)$ 与 $S_2(n)$ 两个部分，分别以 $d$ 和 $\lfloor \frac{n}{d^2}\rfloor$ 来计算。
取一个 $\lfloor \frac{n}{d^2}\rfloor =k\approx n^{\frac{1}{3}}$ 作为分界线。令此时 $d$ 的取值为 $(d_1,d_2]$，则有：
$$\frac{n}{d_1^2}\ge k+1\Rightarrow d_1\le \sqrt{\frac{n}{k+1}}\Rightarrow d_1=\lfloor \sqrt{\frac{n}{k+1}}\rfloor$$
$$\frac{n}{d_2^2}\ge k\Rightarrow d_2=\lfloor \sqrt{\frac{n}{k}}\rfloor$$
于是我们知道了分界线的取值。设 $x_i=\lfloor \sqrt{\frac{n}{i}}\rfloor$，则我们可以知道 $$\begin{gathered} S(n)=S_1(n)+S_2(n)=\sum _{d=1}^{x_{k+1}}\mu (d)\lfloor \frac{n}{d^2}\rfloor +\sum _{i=1}^{n}i\sum _{j=x_{i+1}+1}^{x_i}\mu (j) \\ =\sum _{d=1}^{x_{k+1}}\mu (d)\lfloor \frac{n}{d^2}\rfloor +\sum _{i=1}^{n}i\cdot (M(x_i)-M(x_{i+1})) \end{gathered}$$
于是 $S_1$ 显然可以快速计算了让我们略过，接下来考虑 $S_2$。
首先先计算一下梅滕斯函数 $M(n)={\sum }_{i=1}^n\mu (i)$。
有了例1的经验，显然可以利用 $[n=1]={\sum }_{d|n}\mu (d)$ 来推导：
$$\begin{gathered} 1=\sum _{i=1}^n[i=1]=\sum _{i=1}^n\sum _{d|i}\mu (d) \\ =\sum _{i=1}^n\sum _{d=1}^{\lfloor \frac{n}{i}\rfloor }\mu (d)=\sum _{i=1}^{n}M(\lfloor \frac{n}{i}\rfloor ) \end{gathered}$$
然后又由于这个 $M(x_i)$ 的分布在上面计算 $M(n)$ 的时候较容易出现重复，所以其实你只需要在杜教筛求 $M(n)$ 的时候记忆化就可以了。
最后结论是这个值趋于 $\frac{n}{1-\frac{6}{{\pi }^2}}$，在值域附近二分即可。

代码

//Night's template
//orzakyrt
#include <bits/stdc++.h>
#define R register
#define LL long long
#ifdef __DEBUG__
	#define Debug (fprintf(stderr,"orzyrt\n"),fflush(stderr))
	#define debug(__a...) (fprintf(stderr,__a),fflush(stderr))
	#define setfile(__a) ('orzyrt')
	#define Compile(__a) system("g++ "__a".cpp -lm -o "__a)
#else
	#define Debug ('orzyrt')
	#define debug(__a...) ('orzyrt')
	#define setfile(__a) freopen(__a".in","r",stdin);freopen(__a".out","w",stdout)
	#define Compile(__a) ('orzyrt')
#endif
template<class TT>inline TT Max(R TT a,R TT b){return a<b?b:a;}
template<class TT>inline TT Min(R TT a,R TT b){return a<b?a:b;}
template<class TT>inline TT Abs(R TT a){return a<0?-a:a;}
template<class TT>inline TT cmin(R TT &a,R TT b){(b<a)&&(a=b);}
template<class TT>inline TT cmax(R TT &a,R TT b){(a<b)&&(a=b);}
using namespace std;
template<class TT>inline void read(R TT &x){
	x=0;R bool f=false;R char c=getchar();
	for(;c<48||c>57;c=getchar())f|=(c=='-');
	for(;c>47&&c<58;c=getchar())x=(x<<1)+(x<<3)+(c^48);
	(f)&&(x=-x);
}
//end template

namespace Night{
	const int maxn = 50000010;
	int pr[maxn>>3],*p=pr;
	bool np[maxn];
	int mu2[maxn],mu[maxn];
	map<int,int>mp;
	inline void sieve(){
		mu2[1]=mu[1]=1;
		for(R int i=2;i<maxn;++i){
			if(!np[i])*++p=i,mu[i]=-1;
			for(R int *j=pr+1;j<=p && i**j<maxn;++j){
				np[i**j]=1;
				if(i%*j)mu[i**j]=-mu[i];
				else{mu[i**j]=0;break;}
			}
		}
		for(R int i=2;i<maxn;++i){
			mu2[i]=mu2[i-1]+(mu[i]?1:0);
			mu[i]+=mu[i-1];
		}
	}
	int smu(R int n){
		if(n<maxn)return mu[n];
		if(mp.count(n))return mp[n];
		R int ret=1;
		for(R int i=2,j,k;i<=n;i=j+1){
			k=n/i;j=n/k;
			ret-=smu(k)*(j-i+1);
		}
		return mp[n]=ret;
	}
	inline LL calc(R LL n){
		if(n<maxn)return mu2[n];
		R LL ret=0,lst=0,t,i=1;
		for(;i*i*i<=n;++i,lst=t)ret+=(n/(i*i))*((t=mu[i])-lst);
		for(ret-=(t=n/(i*i))*lst;t;t--)ret+=smu(sqrt(n/t));
		return ret;
	}
	LL n,ans;
	const long double pi=acos(-1);
	int main(){
		sieve();
		read(n);
		R LL l,r,mid;
		l=n/(1-6/pi/pi),r=l+400000,l-=400000;
		l=max(l,1ll);
		while(l<=r){
			mid=l+r>>1;
			if(mid-calc(mid)>=n)ans=mid,r=mid-1;
			else l=mid+1;
		}
		cout<<ans;
		return 0;
	}
}
int main(){
	return Night::main();
}

例3：卷积的构造

求
$$f(n)=\sum _{i=1}^n\sum _{j=1}^i\frac{j}{(i,j)}$$
我们先想办法把 $f(n)$ 化成积性函数前缀和的形式。
令 $A(n)={\sum }_{i=1}^n\frac{i}{(n,i)}$，将其先化简：
$$\begin{gathered} A(n)=\sum _{i=1}^n\frac{i}{(n,i)} \\ =\sum _{i=1}^n\sum _{d|n}[(n,i)=d]\cdot \frac{i}{d} \\ =\sum _{d|n}\sum _{\frac{i}{d}=1}^{\frac{n}{d}}[(\frac{n}{d},\frac{i}{d})=1]\cdot \frac{i}{d} \\ =\frac{1}{2}(1+\sum _{d|n}d\cdot \phi (d)) \end{gathered}$$
于是令函数 $g(n)=2\cdot f(n)-n$，类似之前推的式子，有：
$$\begin{gathered} G(n)=2\cdot F(n)-n \\ =\sum _{i=1}^n\sum _{d|i}d\cdot \phi (d) \\ =\sum _{i=1}^n\sum _{d=1}^{\lfloor \frac{n}{i}\rfloor }d\cdot \phi (d) \end{gathered}$$
因此我们要求一个这样的积性函数的前缀和 $A(n)={\sum }_{i=1}^{n}i\cdot \phi (i)$，我们把 $n$ 表示为质数次幂的形式，有
$$\sum _{d|n}d\cdot \phi (d)=\prod _{i=1}^t\sum _{j=0}^{k_i}{p}_i^j\cdot \phi (p_i^j)=\prod _{i=1}^t\frac{p_i^{2k_i+1}+1}{p_i+1}$$
这并不好计算，但是由于 $(id\cdot \phi )\ast id=i{d}^2$，则（根据上面推的式子）有
$$\begin{gathered} \frac{n\cdot (n+1)\cdot (2n+1)}{6}=\sum _{i=1}^n{i}^2 \\ =\sum _{i=1}^n\sum _{d|i}d\cdot \phi (d)\cdot \frac{i}{d} \\ =\sum _{\frac{i}{d}=1}^n\frac{i}{d}\cdot \sum _{d=1}^{\lfloor \frac{n}{\frac{i}{d}}\rfloor }d\cdot \phi (d) \\ =\sum _{i=1}^{n}i\cdot A(\lfloor \frac{n}{i}\rfloor ) \end{gathered}$$
那么
$$A(n)=\frac{n\cdot (n+1)\cdot (2n+1)}{6}-\sum _{i=2}^{n}i\cdot A(\lfloor \frac{n}{i}\rfloor )$$
于是原问题就可以预处理前 $O(n^{\frac{2}{3}})$ 个以后在 $O(n^{\frac{2}{3}}\log (n))$ 的时间复杂度下解决。
（代码懒得写了）

---

1. $\left[\alpha \right]$ 表示表达式 $\alpha$ 的真值。 ↩︎

2. 证明可以考虑将 $\frac{i}{n}(1\le i\le n)$ 化为最简分数的思路，此处不再赘述。 ↩︎

3. 考虑将莫比乌斯函数视为容斥的系数即可。 ↩︎