整除分块（数论分块）

文章基于b站大佬的视频——整除分块 by泥土笨笨

引例

求

$$\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$$

其中$n\le 10^{12}$，我们打表观察下规律

i	1	2	3	4	5	6	7	8	9	10
$\lfloor \frac{n}{i}\rfloor$	10	5	3	2	2	1	1	1	1	1

以$n=10$为例，表中同样的值会连续出现，而相同的值所划分的区间积是整除分块。整除的性质使得从1到n的$\lfloor \frac{n}{i}\rfloor$可根据数值划分为不同的分块，且分块数远远小于n。利用这种性质，我们如果能推导出每个分块具体的左右端点位置在哪，这个问题就可以快速求解出来了。

整除分块的性质

性质1：$\lfloor \frac{n}{i}\rfloor$最多只有$2\sqrt{n}$种取值

证明：对于$i\le \sqrt{n}$，最多只有$\sqrt{n}$种取值，而对于$i>\sqrt{n}$，有$\frac{n}{i}<\sqrt{n}$，也最多只有$\sqrt{n}$种取值，所以总的值最多不超过$2\sqrt{n}$种

性质2: 若$\lfloor \frac{n}{r}\rfloor$与$\lfloor \frac{n}{i}\rfloor$相等，那么$r$的最大取值为
$$\lfloor \frac{n}{\lfloor \frac{n}{i}\rfloor }\rfloor$$

证明：设$\lfloor \frac{n}{i}\rfloor =k$，于是如果$r$取到比较大的数字，使得$\lfloor \frac{n}{r}\rfloor$与$\lfloor \frac{n}{i}\rfloor$不相等，那么$\lfloor \frac{n}{r}\rfloor <k$，也即$\frac{n}{r}<k$，也即$r>\frac{n}{k}$，那么$r>\lfloor \frac{n}{k}\rfloor$，即$r>\lfloor \frac{n}{\lfloor \frac{n}{i}\rfloor }\rfloor$。所以$r$最大取到$\lfloor \frac{n}{\lfloor \frac{n}{i}\rfloor }\rfloor$可以使得$\lfloor \frac{n}{r}\rfloor$与$\lfloor \frac{n}{i}\rfloor$相等

代码演示：

for(int l = 1;l <= n;l = r + 1) {
	r = n / (n / l);
}

例题

练习1：

已知正整数$n,a,b$，求
$$\sum _{i=1}^n\lfloor \frac{n}{ai+b}\rfloor$$
证明：设$\lfloor \frac{n}{ai+b}\rfloor =k$，于是如果$r$取到比较大的数字，使得$\lfloor \frac{n}{ar+b}\rfloor$与$\lfloor \frac{n}{ai+b}\rfloor$不相等，那么$\lfloor \frac{n}{ar+b}\rfloor <k$，也即$\frac{n}{ar+b}<k$，也即$r>\frac{n-kb}{ak}$，那么$r>\lfloor \frac{n-kb}{ak}\rfloor$。所以$r$最大取到$\lfloor \frac{n-kb}{ak}\rfloor$时$\lfloor \frac{n}{ar+b}\rfloor =\lfloor \frac{n}{ai+b}\rfloor$.

练习2：

P2261 [CQOI2007]余数求和

题目描述

给出正整数 $n$ 和 $k$，请计算

$$G(n,k)=\sum _{i=1}^{n}k\mathrm{mod}i$$

其中 $k\mathrm{mod}i$ 表示 $k$ 除以 $i$ 的余数。

输入格式

输入只有一行两个整数，分别表示 $n$ 和 $k$。

输出格式

输出一行一个整数表示答案。

样例 #1

样例输入 #1

10 5

样例输出 #1

29

提示

样例 1 解释

$G(10,5)=0+1+2+1+0+5+5+5+5+5=29$。

数据规模与约定

• 对于 $30\%$ 的数据，保证 $n,k\le 10^3$。
• 对于 $60\%$ 的数据，保证 $n,k\le 10^6$。
• 对于 $100\%$ 的数据，保证 $1\le n,k\le 10^9$。

解题思路

首先，我们知道
$$k\%i=k-i\lfloor \frac{k}{i}\rfloor$$
所以
$$\sum _{i=1}^{n}kmodi=\sum _{i=1}^n(k-i\lfloor \frac{k}{i}\rfloor )=\sum _{i=1}^{n}k-\sum _{i=1}^{n}i\lfloor \frac{k}{i}\rfloor =nk-\sum _{i=1}^{n}i\lfloor \frac{k}{i}\rfloor$$
而
$$\sum _{i=1}^{n}i\lfloor \frac{k}{i}\rfloor$$
可以用除法分块来求，每次求一个$\lfloor \frac{k}{i}\rfloor$值固定的区间$[l,r]$，这时候把$\lfloor \frac{k}{i}\rfloor$提出来，剩下部分是一个等差数列，直接用公式求和，再乘以$\lfloor \frac{k}{i}\rfloor$即可

代码

#include<iostream>
using namespace std;
typedef long long ll;
int main()
{
    ll n, k, r, l;
    cin >> n >> k;
    ll ans = n * k;
    for (l = 1, r; l <= n; l = r + 1) {
        if (k / l == 0) break;
        r = min(n, k / (k / l));
        ans -= (k / l) * (l + r) * (r - l + 1) / 2;
    }
    cout << ans;
    return 0;
}

练习3：

P6003 [USACO20JAN] Loan Repayment S

题目描述

Farmer John 欠了 Bessie $N$ 加仑牛奶（$1\le N\le 10^{12}$）。他必须在 $K$ 天内将牛奶给 Bessie。但是，他不想将牛奶太早拿出手。另一方面，他不得不在还债上有所进展，所以他必须每天给 Bessie 至少 $M$ 加仑牛奶（$1\le M\le 10^{12}$）。

以下是 Farmer John 决定偿还 Bessie 的方式。首先他选择一个正整数 $X$。然后他每天都重复以下过程：

1. 假设 Farmer John 已经给了 Bessie $G$ 加仑，计算 $\frac{N-G}{X}$ 向下取整。令这个数为 $Y$。
2. 如果 $Y$ 小于 $M$，令 $Y$ 等于 $M$。
3. 给 Bessie $Y$ 加仑牛奶。

求 $X$ 的最大值，使得 Farmer John 按照上述过程能够在 $K$ 天后给 Bessie 至少 $N$ 加仑牛奶 （$1\le K\le 10^{12}$）。

输入格式

输入仅有一行，包含三个空格分隔的正整数 $N,K,M$，满足 $K\times M<N$。

输出格式

输出最大的正整数 $X$，使得按照上述过程 Farmer John 会给 Bessie 至少 $N$ 加仑牛奶。

样例 #1

样例输入 #1

10 3 3

样例输出 #1

2

提示

样例解释

在这个测试用例中，当 $X=2$ 时 Farmer John 第一天给 Bessie $5$ 加仑，后两天每天给 Bessie $M=3$ 加仑。

子任务

• 测试点 $2\sim 4$ 满足 $K\le 10^5$。
• 测试点 $5\sim 11$ 没有额外限制。

解题思路

题目中说求$X$的最大值，那么直接对其二分即可，$X$越大还完牛奶的天数越长，显然满足二分单调性。

对于二分的$check$函数，设$g$是现在已经还的牛奶量，那么今天要还的$y=\frac{n-g}{x}$，假设持续$d$天，每天都还这么多，那么有
$$\lfloor \frac{n-g-yd}{x}\rfloor =y$$
去掉取整符号，也就是
$$\frac{n-g-yd}{x}\ge y$$
整理一下可以得到
$$d\le \frac{n-g-xy}{y}$$
所以$d$最大可以取到
$$\lfloor \frac{n-g-xy}{y}\rfloor$$
注意，当$d=0$时说明只能还今天的，第二天需要还的牛奶量就不一样了，所以实际上分块的大小为$d+1$

代码

#include<iostream>
#include<string>
#include<vector>
#include<cmath>
#include<algorithm>
#include<set>
#include<stack>
#include<map>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
ll n, k, m;
bool check(ll x) {
    ll g = 0, i = 1;
    while (i <= k) {
        ll y = (n - g) / x;
        if (y < m) break;
        ll day = (n - g - x * y) / y;
        // 保持每天还y个，最多持续day天，注意如果day=0意味着还一天
        ++day;
        if (i + day > k) day = k - i + 1;
        i += day;
        g += y * day;
        if (g >= n) return true;
    }
    return (k - i + 1) * m + g >= n;
}
int main()
{
    cin >> n >> k >> m;
    ll l = 1, r = n, ans = 0;
    while (l <= r) {
        ll mid = (r - l) / 2 + l;
        if (check(mid)) ans = mid, l = mid + 1;
        else r = mid - 1;
    }
    cout << ans;
    return 0;
}