[SDOI 2015] 约数个数和（约数个数函数性质 + 莫比乌斯反演） | 错题本

题目

[SDOI 2015] 约数个数和

分析

$$d(xy)=\sum _{d_1|x}\sum _{d_2|y}[\gcd (d_1,d_2)=1]$$ 我们证明每个因子与 $\gcd (d_1,d_2)=1$ 的双射关系：
设 $x={\prod }_{i=1}^n{p_i}^{a_i},y={\prod }_{i=1}^n{p_i}^{b_i}(a_i+b_i\ne 0)$。
考虑 $xy$ 的一个因子 $k={\prod }_{i=1}^n{p_i}^{t_i}$，如果 $t_i\le a_i$，将 $q^{t_i}$ 乘入 $d_1$，否则将 $q^{t_i-a_i}$ 乘入 $d_2$，这样一来显然 $\gcd (d_1,d_2)=1$，也就建立了 $k$ 到 $\gcd (d_1,d_2)=1$ 的单射。再考虑由 $\gcd (d_1,d_2)=1$ 映射回 $k$，也很简单，看每个 $p_i$ 在 $d_1$ 里面还是在 $d_2$ 里面即可，这也是一个单射。
由此我们证明出了每个因子与每个 $\gcd (d_1,d_2)=1$ 之间是一一映射的，原式得证。

---

接下来就是反演了，不妨设 $n\ge m$： $$\begin{array}{rl}& \sum _{i=1}^n\sum _{j=1}^{m}d(ij)\\ =& \sum _{i=1}^n\sum _{j=1}^m\sum _{d_1|i}\sum _{d_2|j}[\gcd (d_1,d_2)=1]\\ =& \sum _{d_1=1}^n\sum _{d_2=1}^m\lfloor \frac{n}{d_1}\rfloor \lfloor \frac{m}{d_2}\rfloor [\gcd (d_1,d_2)=1]\\ =& \sum _{d=1}^m\mu (d)\sum _{d_1=1}^n\sum _{d_2=1}^m[d|d_1][d|d_2]\lfloor \frac{n}{d_1}\rfloor \lfloor \frac{m}{d_2}\rfloor \\ =& \sum _{d=1}^m\mu (d)\sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{d}\rfloor }\lfloor \frac{n}{d\cdot i}\rfloor \lfloor \frac{m}{d\cdot j}\rfloor \\ =& \sum _{d=1}^m\mu (d)\left(\sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }\lfloor \frac{\lfloor \frac{n}{d}\rfloor }{i}\rfloor \right)\left(\sum _{j=1}^{\lfloor \frac{m}{d}\rfloor }\lfloor \frac{\lfloor \frac{m}{d}\rfloor }{j}\rfloor \right)\end{array}$$ 令 $f(x)=\sum _{i=1}^x\lfloor \frac{x}{i}\rfloor$，它可以 $O(n\sqrt{n})$ 初始化，那么原式也可以 $O(\sqrt{n})$ 求了。

代码

#include <bits/stdc++.h>

typedef long long LL;

const int MAXN = 50000;

LL Num[MAXN + 5];

int Mu[MAXN + 5];
bool Vis[MAXN + 5];
std::vector<int> Primes;

void Init(int n) {
	Mu[1] = 1;
	for (int i = 2; i <= n; i++) {
		if (!Vis[i])
			Mu[i] = -1, Primes.push_back(i);
		for (int j = 0; j < (int)Primes.size() && (LL)i * Primes[j] <= n; j++) {
			Vis[i * Primes[j]] = true;
			if (i % Primes[j] == 0) {
				Mu[i * Primes[j]] = 0;
				break;
			}
			Mu[i * Primes[j]] = -Mu[i];
		}
        Mu[i] = Mu[i - 1] + Mu[i];
	}
	for (int i = 1; i <= n; i++) {
        for (int lft = 1, rgt = 1; lft <= i; lft = rgt + 1) {
            rgt = i / (i / lft);
            Num[i] += (LL)(rgt - lft + 1) * (i / lft);
        }
	}
}

int main() {
    Init(MAXN);
    int T; scanf("%d", &T);
    while (T--) {
        int N, M; scanf("%d%d", &N, &M);
        if (N < M) std::swap(N, M);
        LL Ans = 0;
        for (int lft = 1, rgt = 1; lft <= M; lft = rgt + 1) {
            rgt = std::min(M / (M / lft), N / (N / lft));
            Ans += (LL)(Mu[rgt] - Mu[lft - 1]) * Num[N / lft] * Num[M / lft];
        }
        printf("%lld\n", Ans);
    }
    return 0;
}