数论分块学习笔记

一、问题引入

试快速计算下式：
$${\Sigma }_{i=1}^{n}f(i)g(\lfloor \frac{n}{i}\rfloor )$$

其中，$f(x)$ 的前缀和已预处理（或可预处理），$g(x)$ 可以 $O(1)$ 计算。

二、分析

首先，我们可以发现，在 $i$ 从 $1$ 到 $n$ 单调递增的过程中， $\lfloor \frac{n}{i}\rfloor$ 的值从 $n$ 到 $1$ 单调递减，但变化次数远远少于 $n$ 次，因此可以考虑将 $\lfloor \frac{n}{i}\rfloor$ 相同的情况合并，统一计算。更换枚举方式，简化为下式：
$${\Sigma }_{i=1}^{n}f(i)g(\lfloor \frac{n}{i}\rfloor )={\Sigma }_{j\in S}(sumf(r_j)-sumf(l_j))g(j),S=\{\lfloor \frac{n}{i}\rfloor \mid i\in {\mathbb{N}}_{+},i⩽n\}$$

经过简化，从 $1$ 到 $n$ 的枚举范围降低到了枚举 $S$ 集合。那么 $S$ 集合究竟有多大呢？有引理如下。

Lemma. $\forall n\in {\mathbb{N}}_{+},\left|\left\{\lfloor \frac{n}{i}\rfloor \mid i\in {\mathbb{N}}_{+},i⩽n\right\}\right|⩽2\sqrt{n}，$$\left|S\right|$ 表示集合 $S$ 的大小。

简略证明如下：
当 $i$ 小于 $\sqrt{n}$ 时，$\lfloor \frac{n}{i}\rfloor$ 至多有 $\sqrt{n}$ 种取值。
当 $i$ 大于 $\sqrt{n}$ 时，$1⩽\lfloor \frac{n}{i}\rfloor ⩽\sqrt{n}$ ，也至多有 $\sqrt{n}$ 种取值。
因此，$\lfloor \frac{n}{i}\rfloor$ 总共至多 $2\sqrt{n}$ 种取值。

最后，分析一下更换枚举后的复杂度。枚举至多 $2\sqrt{n}$ 次，单次计算区间和和 $g(x)$ 函数值均可以在 $O(1)$ 的时间复杂度内解决，因此总复杂度为 $O(\sqrt{n})$。

三、例题和代码

（1）UVA11526

题目大意：$T$ 组数据，每组给定一个 $n$ ，求下式的值。
$${\Sigma }_{i=1}^n\lfloor \frac{n}{i}\rfloor$$

解：数论分块板子题。其中 $f(x)\equiv 1,g(x)=x$。代码如下。

#include<bits/stdc++.h>
using namespace std;
int T, n;
int sumf(int x){
	return x;
}
int g(int x){
	return x;
}
int main(){
	scanf("%d", &T);
	while (T --){
		scanf("%d", &n);
		long long l = 1, r, ans = 0;
		while (l <= n){
			r = n / (n / l);
			ans += (sumf(r) - sumf(l - 1)) * g(n / l);
			l = r + 1;
		}
		printf("%lld\n", ans);
	}
	return 0;
}

(2) P2261

题目大意：给定 $n$ 和 $k$ ，求下式的值。
$$G(n,k)={\Sigma }_{i=1}^{n}k\%i$$

其中 $\%$ 为取余运算。
解：转化。
$$G(n,k)={\Sigma }_{i=1}^{n}k\%i={\Sigma }_{i=1}^n(k-i\lfloor \frac{k}{i}\rfloor )={\Sigma }_{i=1}^{n}k-{\Sigma }_{i=1}^{n}i\lfloor \frac{k}{i}\rfloor$$
与数论分块的模板式子很相似，可以确定两个函数分别为 $f(x)=x,g(x)=x$ 。不过上界不同，分类讨论即可。
对于 $k⩽n$的情况，由于 $i⩾k$ 时 $\lfloor \frac{k}{i}\rfloor =0$，没有影响。
对于 $k>n$ 的情况，需要判断区间的右端点 $r$ 是否超过 $n$ ，如果超过，则需要修改。代码如下。

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