【算法讲6：数论分块（整除分块）】引入 | 思路证明 | 例题

引入

• 给你一个正整数 $n$ ，让你求
  $$\underset{i=1}{\sum ^n}\lfloor \frac{n}{i}\rfloor$$
  其中 $1\le n\le 10^9$，$\lfloor x\rfloor$ 表示 $x$ 向下取整，$\underset{i=1}{\sum ^n}f(i)$ 表示求 $i=1,2,\cdots ,n$时候，$f(i)$ 的和。

思路

• 我们首先考虑，暴力 $for$ 循环做的话，时间复杂度为 $O(n)$，会超时肯定不行。
• 我们其次考虑打表，看看有什么规律。当 $n=30$ 时候：

i	1	2	3	4	5	6	7	8	9	10	11	12	13	14	15	16	17	18	19	20	21	22	23	24	25	26	27	28	29	30
$\lfloor \frac{n}{i}\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

• 可以看到，对于某一段 $i$ ， $\lfloor \frac{n}{i}\rfloor$ 的值都相同。
  我们能不能这样一段一段地枚举，然后分别对每一段计算贡献呢？
• 如果我们知道了每一段的左端点 $L$ 和右端点 $R$
  那么这一段的贡献容易得到：$(R-L+1)\times \lfloor \frac{n}{L}\rfloor$
  并且我们也可以轻松得到下一段的左端点为目前右端点+1
• 那么最难问题来了：已知左端点 $L$ ，怎么求这一段的右端点 $R$ ？
  转变成数学语言，就是已知 $L$ ，求 $R_{\max }$，使得 $\lfloor \frac{n}{L}\rfloor =\lfloor \frac{n}{R}\rfloor$
  我们设 $\lfloor \frac{n}{L}\rfloor =k$，即 $n=Lk+p$，$0\le p<L$
  我们设 $R=L+d$，带入得到 $\lfloor \frac{n}{L+d}\rfloor =k$，即 $n=(L+d)k+p^{\prime }$，$0\le p^{\prime }<L$
  下面式子减去上面式子，得到 $p^{\prime }=p-dk$
  此时 $d_{\max }=\lfloor \frac{p}{k}\rfloor$，带入 $R=L+d$ 得到
  $$\begin{array}{rl}R& =L+\lfloor \frac{p}{k}\rfloor \\ & =L+\lfloor \frac{n\%L}{\lfloor \frac{n}{L}\rfloor }\rfloor \\ & =L+\lfloor \frac{n-\lfloor \frac{n}{L}\rfloor L}{\lfloor \frac{n}{L}\rfloor }\rfloor \\ & =\lfloor L+\frac{n-\lfloor \frac{n}{L}\rfloor L}{\lfloor \frac{n}{L}\rfloor }\rfloor \\ & =\lfloor \frac{\lfloor \frac{n}{L}\rfloor L}{\lfloor \frac{n}{L}\rfloor }+\frac{n-\lfloor \frac{n}{L}\rfloor L}{\lfloor \frac{n}{L}\rfloor }\rfloor \\ & =\lfloor \frac{n}{\lfloor \frac{n}{L}\rfloor }\rfloor \end{array}$$
  我们得到了，对于某一段的 $L$ ，这时$R=\lfloor \frac{n}{\lfloor \frac{n}{L}\rfloor }\rfloor$
• 最后一个问题：该算法的时间复杂度是多少？
  易得，对于一个数 $n$ ，该数字的正因子的个数是 $O(2\sqrt{n})$ 级别的
  故数论分块的时间复杂度为 $O(\sqrt{n})$

例题

约束研究 | 洛谷 P1403

• [AHOI2005]约数研究
  求 $\underset{i=1}{\sum ^n}d(i)$
  $1\le n\le 10^6$，其中 $d(n)$ 表示 $n$ 的约数个数。

---

• 稍微想一下，枚举每一个数的约数的个数很复杂，那我们直接枚举每一个约数，考虑每个约数的贡献次数。
  约数范围 $1\sim n$，对于某一个约数 $i$ ，共有 $\lfloor \frac{n}{i}\rfloor$ 个数字有该约数。
  式子直接转化为 $\underset{i=1}{\sum ^n}\lfloor \frac{n}{i}\rfloor$
  注意到，我们可以做 $n\le 10^{14}$ 的数据范围。

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
ll solve(ll x){
    ll L = 1,R = 0;
    ll res = 0;
    while(L <= x){
        R = x/(x/L);
        res += (x / L) * (R - L + 1);
        L = R + 1;
    }
    return res;
}

int main()
{
    ll n;cin >> n;
    cout << solve(n);
    return 0;
}

---

约数和 | 洛谷 P2424

• 约数和 | 洛谷 P2424
  给你正整数 $a、b$ ，求 $\underset{i=a}{\sum ^b}\sigma (i)$
  $1\le a\le b\le 2\times 10^9$，其中 $\sigma (n)$ 表示 $n$ 的约数和。

---

• 容易想到，答案即为 $\underset{i=1}{\sum ^b}\sigma (i)-\underset{i=1}{\sum ^{a-1}}\sigma (i)$
  我们要求 $\underset{i=1}{\sum ^n}\sigma (i)$，枚举每一个约数，计算答案，式子转化为
  $\underset{i=1}{\sum ^n}i\times \lfloor \frac{n}{i}\rfloor$
  对于同一段，计算贡献为 $L\times (L+1)\times \cdots \times R\times \lfloor \frac{n}{L}\rfloor =\lfloor \frac{n}{L}\rfloor \times \frac{(L+R)\times (R-L+1)}{2}$

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
ll solve(ll n){
    ll L = 1,R = 0;
    ll res = 0;
    while(L <= n){
        R = n/(n/L);
        res += n / L * (L + R) * (R - L + 1) / 2;
        L = R + 1;
    }
    return res;
}

int main()
{
    ll a,b;cin >> a >> b;
    cout << solve(b) - solve(a-1);
    return 0;
}

---

余数求和 | 洛谷 P2261

• 余数求和 | 洛谷 P2261
  给你正整数 $n、k$，求 $\underset{i=1}{\sum ^n}k\%i$
  $1\le n，k\le 10^9$

---

• 取模符号？我们化简一下式子
  $\underset{i=1}{\sum ^n}k\%i=\underset{i=1}{\sum ^n}k-i\lfloor \frac{k}{i}\rfloor =nk-\underset{i=1}{\sum ^n}i\lfloor \frac{k}{i}\rfloor$
  看到，右半部分类似一个整除分块。不同的是，分子部分为 $k$ 而不是 $n$。
  考虑到 $\forall x>k，都有\lfloor \frac{k}{x}\rfloor =0$。
  并且我们每次枚举，要保证右端点 $R$ 不超过 $n$ 。

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
ll solve(ll n,ll k){
    ll L = 1,R = 0;
    ll res = n * k;
    while(L <= min(n,k)){
        R = min(n,k/(k/L));
        res -= (k / L) * (L + R) * (R - L + 1) / 2;
        L = R + 1;
    }
    return res;
}

int main()
{
    ll n,k;cin >> n >> k;
    cout << solve(n,k);
    return 0;
}

${}^{\star }$扩展$\lceil$二维数论分块$\rfloor$

• 求 $\underset{i=1}{\sum ^{\min (n,m)}}\lfloor \frac{n}{i}\rfloor \times \lfloor \frac{m}{i}\rfloor$
• 思路：类似于一维，只不过二维的块数变为 $2\sqrt{m}+2\sqrt{n}$块。
  每一个小块里面，$\lfloor \frac{n}{i}\rfloor \times \lfloor \frac{m}{i}\rfloor$ 的值都相同。
  我们对于一个 $L$ ，最大的 $R$ 怎么找呢？易得为 min ⁡ { ⌊ n ⌊ n L ⌋ ⌋ , ⌊ m ⌊ m L ⌋ ⌋ } \min\{\lfloor\frac{n}{\lfloor\frac{n}{L}\rfloor}\rfloor,\lfloor\frac{m}{\lfloor\frac{m}{L}\rfloor}\rfloor\} min{⌊⌊Ln​⌋n​⌋,⌊⌊Lm​⌋m​⌋}

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
ll solve(ll n,ll m){
    ll L = 1,R = 0;
    ll res = 0;
    while(L <= min(n,m)){
        R = min(m/(m/L),n/(n/L));
        res += (n/L) * (m/L) * (R - L + 1);
        L = R + 1;
    }
    return res;
}

模积和 | [清华集训2012] | 洛谷 P 2260

• 求 $\underset{i=1}{\sum ^n}\underset{j=1}{\sum ^m}(n\mathrm{m}\mathrm{o}\mathrm{d}i)\times (m\mathrm{m}\mathrm{o}\mathrm{d}j)，i\ne j$
• 思路：不要忘记 $i\ne j$ 啊！
  我们开始推狮子：
  $\begin{array}{rl}I& =\underset{i=1}{\sum ^n}\underset{j=1}{\sum ^m}(n-\lfloor \frac{n}{i}\rfloor i)\times (m-\lfloor \frac{m}{j}\rfloor j)-\underset{i=1}{\sum ^{\min (n,m)}}(n-\lfloor \frac{n}{i}\rfloor i)\times (m-\lfloor \frac{m}{i}\rfloor i)\\ & =\underset{i=1}{\sum ^n}(n-\lfloor \frac{n}{i}\rfloor i)\underset{j=1}{\sum ^m}(m-\lfloor \frac{m}{j}\rfloor j)-\underset{i=1}{\sum ^{\min (n,m)}}(nm-n\lfloor \frac{m}{i}\rfloor i-m\lfloor \frac{n}{i}\rfloor i+\lfloor \frac{m}{i}\rfloor \lfloor \frac{n}{i}\rfloor i^2)\\ & =(n^2-\underset{i=1}{\sum ^n}\lfloor \frac{n}{i}\rfloor i)(m^2-\underset{j=1}{\sum ^m}\lfloor \frac{m}{j}\rfloor j)-\min (n,m)\times nm+n\underset{i=1}{\sum ^{\min (n,m)}}\lfloor \frac{m}{i}\rfloor i+m\underset{i=1}{\sum ^{\min (n,m)}}\lfloor \frac{n}{i}\rfloor i-\underset{i=1}{\sum ^{\min (n,m)}}\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{i}\rfloor i^2\end{array}$
  好家伙，清华集训的题就是不一样。。。
  注意：模数 $19940417=7\times 2848631$不是质数！用拓欧或者欧拉定理求出逆元！
  我们使用函数封装一下，即可优雅 地敲出AC代码
  时间复杂度：$O(\sqrt{\max (n,m)})$

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
const ll  MOD = 19940417;

ll ex_gcd(ll a,ll b,ll& x,ll& y)
{
    if(b==0)
    {
        x=1;y=0;
        return a;
    }
    ll ans=ex_gcd(b,a%b,x,y);
    ll tmp=x;
    x=y;
    y=tmp-a/b*y;
    return ans;
}
ll iv(ll a,ll mod)//存在逆元条件：gcd(a,mod)=1
{
    ll x,y;
    ll g=ex_gcd(a,mod,x,y);
    if(g!=1)return -1;
    return (x % mod + mod) % mod;
}

///	数论分块，项数为 n ，分子为 k
ll solve(ll n,ll k){
    ll L = 1,R = 0;
    ll res = 0;
    while(L <= min(n,k)){
        R = min(n,k/(k/L));
        res = (res + k / L % MOD * (L + R) % MOD * (R - L + 1) % MOD * iv(2,MOD) % MOD);
        L = R + 1;
    }
    res = (res % MOD + MOD) % MOD;
    return res;
}
/// 快速算 1^2 + 2^2 + …… + x^2
ll pre(ll x){
    ll res = x * (x + 1) % MOD * (2 * x + 1) % MOD * iv(6,MOD) % MOD;
    return res;
}
///	二维数论分块，项数为 n ，一项分子为 a，另一项分子为 b
ll solve2(ll n,ll a,ll b){
    ll L = 1,R = 0;
    ll res = 0;
    while(L <= n){
        R = min(a/(a/L),b/(b/L));
        res = (res + (a / L) * (b / L) % MOD * (pre(R) - pre(L - 1) % MOD)) % MOD;	/// 注意 i^2 快速求和
        L = R + 1;
    }
    res = (res % MOD + MOD) % MOD;
    return res;
}
int main()
{
    ll n,m;
    cin >> n >> m;
    ll ans = (n * n % MOD - solve(n,n)) * (m * m % MOD - solve(m,m)) % MOD;
    ans = (ans - min(n,m) * n % MOD * m % MOD + n * solve(min(n,m),m) % MOD + m * solve(min(n,m),n) % MOD) % MOD;
    ans = (ans - solve2(min(n,m),n,m)) % MOD;
    ans = (ans + MOD) % MOD;
    cout << ans;
    return 0;
}

${}^{\star }$扩展$\lceil$数论分块变形$\rfloor$

• 求 $\underset{i=1}{\sum ^n}\lceil \frac{n}{i}\rceil$
  其中 $\lceil x\rceil$ 表示 $x$ 向上取整。
• 思路：有一个转化：对于正整数$a、b$，我们有 $\lceil \frac{a}{b}\rceil =\lfloor \frac{a+b-1}{b}\rfloor$ 。
  证明：设 $\lceil \frac{a}{b}\rceil =\lfloor \frac{a+b-1}{b}\rfloor =x$。
  因为 $\lceil \frac{a}{b}\rceil =x$，我们得出 $\frac{a}{b}\in (x-1,x]$
  又因为 $b$ 是正整数，所以 $\frac{b-1}{b}\in [0,1)$
  所以 $\frac{a}{b}+\frac{b-1}{b}\in (x-1,x+1)$
  所以 $\lfloor \frac{a+b-1}{b}\rfloor =x$
  所以$\lceil \frac{a}{b}\rceil =\lfloor \frac{a+b-1}{b}\rfloor \square$
• 代替原式子，得到 $\underset{i=1}{\sum ^n}\lceil \frac{n}{i}\rceil =\underset{i=1}{\sum ^n}\lfloor \frac{n+i-1}{i}\rfloor =\underset{i=1}{\sum ^n}(\lfloor \frac{n-1}{i}\rfloor +1)$
  这个应该不用代码了吧，和上述代码同理请仔细思考哦。