[数论数学]反演卷积的应用-快速求和化简

可以说是卷积和反演的综合应用

一、整除分块（数论分块）

补充一点基础知识…

具体来说就是：如果我们要求$\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$，并不需要用朴素算法$O(n)$去求，可以加快速度。

定理1：所有$1\le i\le n$的$\lfloor \frac{n}{i}\rfloor$的结果最多只有$2\sqrt{n}$种。

证明：对于$i\in [1,\sqrt{n}]$，显然最多只有$\sqrt{n}$种取值，而对于$i\in [\sqrt{n}+1,n]$，$\lfloor \frac{n}{i}\rfloor \le \sqrt{n}$，因此也最多有$\sqrt{n}$种取值，两个部分相加，因此最多有$2\sqrt{n}$种取值。

定理2：对于确定的$i$，能使得$\lfloor \frac{n}{x}\rfloor =\lfloor \frac{n}{i}\rfloor$的$x_{max}=\lfloor \frac{n}{\lfloor \frac{n}{i}\rfloor }\rfloor$。

这里选取了两种证明方法。

证法1：记$f(i)=\lfloor \frac{n}{\lfloor \frac{n}{i}\rfloor }\rfloor$，则因为$\lfloor \frac{n}{i}\rfloor$单调不减，而且$f(i)\ge i$，所以必有$\lfloor \frac{n}{f(i)}\rfloor \le \lfloor \frac{n}{i}\rfloor$，又因为$f(i)\le \frac{n}{\lfloor \frac{n}{i}\rfloor }$，所以有：

$\lfloor \frac{n}{f(i)}\rfloor \ge n/(\frac{n}{\lfloor \frac{n}{i}\rfloor })=\lfloor \frac{n}{i}\rfloor$

同时证明大于等于和小于等于，因此$\lfloor \frac{n}{f(i)}\rfloor =\lfloor \frac{n}{i}\rfloor$。

另外，若$f(i)$不是满足条件的最大值，则$\lfloor \frac{n}{f(i)+1}\rfloor =\lfloor \frac{n}{i}\rfloor$，则$f(i)=f(f(i)+1)$，而又$f(i)\ge i$，矛盾，因此$f(i)$就是满足要求的最大值。

---

证法2：记$n=di+p=dx+q(0\le p,q<n)$，则$\lfloor \frac{n}{x}\rfloor =\lfloor \frac{n}{i}\rfloor =d$，由前式两等式相减得：

$q=p-d(x-i)q=p-dk(k=x-i)$

要使$x$最大，也就是要让$k$尽量大，由$q\ge 0$可得$k$的最大值为$k_{max}=\lfloor \frac{p}{d}\rfloor$。

此时有：$x=i+d_{max}=i+\lfloor \frac{p}{d}\rfloor$，将$p,d$代换得：$x=i+\lfloor \frac{nmodi}{\lfloor \frac{n}{i}\rfloor }\rfloor$，将$nmodi$代换，并将$i$乘上去可得：

$x=i+\lfloor \frac{n-i\lfloor \frac{n}{i}\rfloor }{\lfloor \frac{n}{i}\rfloor }\rfloor =\lfloor \frac{i\lfloor \frac{n}{i}\rfloor +n-i\lfloor \frac{n}{i}\rfloor }{\lfloor \frac{n}{i}\rfloor }\rfloor =\lfloor \frac{n}{\lfloor \frac{n}{i}\rfloor }\rfloor$

同样得到了结论。

---

因此我们只要通过上面对商的讨论，就可以在$O(\sqrt{n})$时间复杂度内快速计算$\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$。

for (int l=1,r;l<=n;l=r+1) {
   
	r=n/(n/l);
	ans+=(n/l)*(r-l+1);
}

来看一道题目。

题目大意：给定$n,k\le 10^9$，求出$\sum _{i=1}^{n}kmodi=kmod1+...+kmodn$。洛谷2261

分析：直接$O(n)$做是肯定不行的，这时我们想到利用余数的定义：$nmodi=n-i\lfloor \frac{n}{i}\rfloor$，于是我们就可以开心化简了：

$\sum _{i=1}^{n}kmodi=kn-\sum _{i=1}^{n}i\lfloor \frac{k}{i}\rfloor$

而这个式子的后面一部分非常像整除分块的形式，但是还不完全一样。不过这也不难，我们只要将上面的代码中原本表示$l-r$之间数的个数的$(r-l+1)$替换成$l-r$的和$\frac{(l+r)(r-l+1)}{2}$即可。

有一点需要注意，此时求和上限和被除数不同，需要加上一句额外判断，当$\lfloor \frac{k}{\lfloor \frac{k}{i}\rfloor }\rfloor$超过了$n$时也只能取到$n$。

#include <bits/stdc++.h>
using namespace std;
long long ans,n,k;
int main () {
   
	scanf("%lld%lld",&n,&k);
	ans=n*k;
	for (long long l=1,r;l<=min(n,k);l=r+1) {
   
		r=min(k/(k/l),n);
		ans-=(k/l)*(l+r)*(r-l+1)/2;
	}
	printf("%lld\n",ans);
	return 0;
}

通过上面这道题目我们能看出来什么？其实整除分块并不只能求原本的标准形式。事实上，形如整除式后面乘上一个函数的值这样的求和式，都可以用整除分块的方法做出，即：

$\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor f(i)=\sum \lfloor \frac{n}{l}\rfloor (s(r)-s(l-1))$

其中$s(k)=\sum _{i=1}^{k}f(i)$，即函数的前缀和，$l,r$与上面代码中的意义相同。其实上面的例子也是这样，只是$s(i)$恰好是一个等差数列的形式，很容易算出。

---

二、gcd快速求和

由于这个部分比较好玩，而且与反演和卷积也有点关系，体现一些基本思想，就先讲一些了。

一维gcd求和：求$\sum _{i=1}^{N}gcd(N,i)$。洛谷2303

分析：我们对不同的数据范围进行逐层递进的分析：

（1）$N\le 10^5$

这个非常简单，只要你会欧几里得算法，就可以直接计算$gcd(N,i)$的值，然后将所有结果相加，暴力计算即可得到答案，时间复杂度$O(N\log N)$；

（2）$N\le 10^9$

主要的思维都体现在这里了…首先处理这种最大公因数问题的最常用方法之一，是枚举最大公因数的值$d$，也就是我们枚举每一个可能的$d$，然后去计数有多少个为$d$的结果，将$d$与计数量相乘即可得到结果为$d$的最大公因数和了，而$d=gcd(N,i)$，则显然$d|N$，因此可以这样枚举：

$\sum _{d|N}d\sum _{i=1}^N[gcd(N,i)=d]$

这是第一步技巧，将gcd提取出来放在外面。

接下来还要继续化简，我们现在感觉这个方括号里的等式真的没什么用，那么遇到这种情况有两种方法，一种是用后面要讲的莫比乌斯反演，但这里还不必要；另一种就是“约分”，也就是将内侧求和的每一个d都约去，这样的话$N$就变成了$\frac{N}{d}$了：

$\sum _{d|N}d\sum _{i=1}^{\frac{N}{d}}[gcd(\frac{N}{d},i)=1]$

这时候你就应该意识到了：此时里面这个求和号其实就是欧拉函数！我们将它表示一下：

$\sum _{d|N}d\phi (\frac{N}{d})$

这样做$10^9$数据就足够了，我们枚举N的因子d，并算出N/d的欧拉函数值就可以了。

（3）$N\le 10^{12}$

这时上面计算欧拉函数的时间代价就太高了，因此需要再次优化。

如果你掌握之前讲的卷积足够好的话，你就应该及时意识到，上面的这个式子是一个卷积式，其实是$id$与$\phi$的卷积。之前我们还学过：两个积性函数的卷积依然是积性函数，所以此时我们假设$f(N)=\sum _{d|N}d\phi (\frac{N}{d})$，此时$f$就是一个积性函数。那么我们将$n$分解素因数为$N=p_1^{c_1}...p_n^{c_n}$，则：