「ACM/OI」【数论】Update:1.18

数论分块

介绍

• 数论分块可以快速计算一些含有除法向下取整的和式（即形如 ${\sum }_{i=1}^{n}f(i)g(\lfloor \frac{n}{i}\rfloor )$ 的和式）。当可以在 $O(1)$ 内计算 $f(r)-f(l)$ 或已经预处理出 $f$ 的前缀和时，数论分块就可以在 $O(\sqrt{n})$ 的时间内计算上述和式的值。

引入

• 思考如何求解 ${\sum }_{i=1}^n\lfloor \frac{n}{i}\rfloor$，$1\le n\le 10^{12}$
• 暴力 $O(n)$ 枚举显然是不行的，但如果一时没有思路，不妨先尝试通过暴力发现一些什么：
• 对于 $n=11$， ⌊ n i ⌋ = { 11 , 5 , 3 , 2 , 2 , 1 , 1 , 1 , 1 , 1 , 1 } \left\lfloor\dfrac ni\right\rfloor = \{ 11,5,3,2,2,1,1,1,1,1,1 \} ⌊in​⌋={11,5,3,2,2,1,1,1,1,1,1}
• 对于 $n=30$， ⌊ n i ⌋ = { 30 , 15 , 10 , 7 , 6 , 5 , 4 , 3 , 3 , 3 , 2 , 2 , 2 , 2 , 2 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 } \left\lfloor\dfrac ni\right\rfloor = \{30,15,10,7,6,5,4,3,3,3,2,2,2,2,2,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1 \} ⌊in​⌋={30,15,10,7,6,5,4,3,3,3,2,2,2,2,2,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1}
• 显而易见，数列中有几段数值相同的连续部分，即对于同一个 $n$，有不少 $i$ 使得 $\lfloor \frac{n}{i}\rfloor$ 的值相同，没有必要多次计算，于是整除分块便出现了。如何快速计算每一个数值对应的区间呢？
• 对于枚举的每个区间的左端点 $l$，首先易得该区间的数值均为 $\lfloor \frac{n}{l}\rfloor$，由结论可知区间右端点 $r=\lfloor \frac{n}{\lfloor \frac{n}{l}\rfloor }\rfloor$

时间复杂度

• $O(\sqrt{n})$. 可以证明块数不大于 $2\sqrt{n}$。简单来说可以分类讨论一下，若 $i\le \sqrt{n}$，$i$ 最多可以有 $\sqrt{n}$ 种取值，因此 $\lfloor \frac{n}{i}\rfloor$ 最多有 $\sqrt{n}$ 种取值；若 $i>\sqrt{n}$，显然 $\lfloor \frac{n}{i}\rfloor$ 最多有 $\sqrt{n}$ 种取值。综上，$\lfloor \frac{n}{i}\rfloor$ 最多有 $2\sqrt{n}$ 种取值。整个算法时间复杂度为 $O(\sqrt{n})$。

例题1：约数个数

题目描述：
设 $\sigma (x)$ 表示 $x$ 的约数个数，给定 $n$$(1\le n<2^{32})$，求 ${\sum }_{i=1}^n\sigma (i)$

分析

• 数学语言：${\sum }_{i=1}^n\sigma (i)={\sum }_{i=1}^{n}sum(d|i)={\sum }_{d=1}^n\lfloor \frac{n}{d}\rfloor$
• 也就是说，相较于朴素思想即统计 $[1,n]$ 每个数的约数个数，我们转化为考虑每一个约数 $d\in [1,n]$ 在区间中的总贡献（某个约数在所有数的约数集合中出现的总次数），对于总贡献相同的区间（某些约数的出现次数相同），答案累加 区间长度乘上出现次数 即可。

代码

void solve() {
	ll n; cin >> n;
	ll ans = 0;
	for(ll l=1,r;l<=n;l=r+1) { // 枚举区间左端点
		r = n / (n / l);       // 计算区间右端点
		ll h = n / l;          // 该区间约数的相同出现次数
		ans += h*(r-l+1);
	}
	cout << ans;
}

例题2：约数和

题目描述：
设 $\sigma (x)$ 表示 $x$ 所有约数的和，给定 $x,y$$(1\le x<y\le 2\times 10^9)$，求 ${\sum }_{i=x}^y\sigma (i)$

分析

• 数学语言：${\sum }_{i=1}^n\sigma (i)={\sum }_{i=1}^n{\sum }_{d|x}d={\sum }_{d=1}^{n}d\times \lfloor \frac{n}{d}\rfloor$
• 基本思想与例题1大同小异，注意记录的是约数的和，考虑 $[l,r]$ 约数区间和。
• 另外本题求解的是 $[x,y]$ 中数的约数和，利用前缀和的思想分别求解 $f_y$ 和 $f_{x-1}$，作差即为答案。

代码

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
ll solve(int n) {
	ll ans = 0;
	for(ll l=1,r;l<=n;l=r+1) {
		r = n / (n / l);
		ll h = n / l;
		ans += h*(l+r)*(r-l+1)/2; // 等差数列求和
	}
	return ans;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	int x,y; cin >> x >> y;
    cout << solve(y)-solve(x-1) << "\n";
	return 0;
} 

例题3：余数求和

题目描述：
给定正整数 $n,k$，求 $G(n,k)={\sum }_{i=1}^{n}k\mathrm{mod}i$

分析

• 对于取模运算，可以转化：${\sum }_{i=1}^{n}k\mathrm{mod}i={\sum }_{i=1}^{n}k-i\times \lfloor \frac{k}{i}\rfloor =n\times k-{\sum }_{i=1}^{n}i\times \lfloor \frac{k}{i}\rfloor$
• 对于后面一项，我们发现它与 ${\sum }_{i=1}^{n}f(i)g(\lfloor \frac{n}{i}\rfloor )$ 形式上十分接近，可以考虑整除分块，不同的是分子部分为 $k$ 而不是 $n$。
• 考虑 $n>k$，对于 $\forall i>k$，有 $\lfloor \frac{k}{i}\rfloor =0$，故 ${\sum }_{i=1}^{n}i\times \lfloor \frac{k}{i}\rfloor ={\sum }_{i=1}^{k}i\times \lfloor \frac{k}{i}\rfloor +{\sum }_{i=k+1}^{n}i\times \lfloor \frac{k}{i}\rfloor ={\sum }_{i=1}^{k}i\times \lfloor \frac{k}{i}\rfloor +0={\sum }_{i=1}^{k}i\times \lfloor \frac{k}{i}\rfloor$，所以左边界枚举到 $n,k$ 的较小值即可。同时，右边界不能超过 $n$。

代码

void solve() {
	ll n, k; cin >> n >> k;
	ll res = 0;
	for(ll l=1,r;l<=min(n,k);l=r+1) {
		r = min(n,k/(k/l));
		ll h = k / l;
		res += h * (l+r)*(r-l+1)/2;
	}
	cout << n*k-res << "\n";
}