【luogu P3327】约数个数和（莫比乌斯反演）

约数个数和

题目链接：luogu P3327

题目大意

给你 n,m 要你求 ∑i=1~n∑j=1~m d(ij)，d 是约数个数。
多组数据。

思路

首先由一个比较神奇的东西：
$d(ij)=\sum _{x|i}\sum _{y|j}[gcd(x,y)=1]$

这里证明一下：
先看我们可以只看一个质数的部分。
假设它的因子的出现次数是 $x$，$i$ 中有 $a$ 个，$j$ 中有 $b$ 个，那 $ij$ 中就是 $a+b$ 个。

如果 $x⩽a$，那它在 $i$ 中会出现被数到。
那如果 $x>a$，我们可以在 $j$ 中数一个 $x-a$ 就表示满了之后再我那个这边的个数。

所以是对的。

然后就来化式子：
$\sum _{i=1}^n\sum _{j=1}^{m}d(ij)$
$=\sum _{i=1}^n\sum _{j=1}^m\sum _{x|i}\sum _{y|j}[gcd(x,y)=1]$
$=\sum _{x=1}^n\sum _{y=1}^m\sum _{i=1}^{\frac{n}{x}}\sum _{j=1}^{\frac{m}{y}}[gcd(x,y)=1]$
$=\sum _{x=1}^n\sum _{y=1}^m\lfloor \frac{n}{x}\rfloor \lfloor \frac{m}{y}\rfloor [gcd(x,y)=1]$
$=\sum _{i=1}^n\sum _{j=1}^m\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{j}\rfloor [gcd(i,j)=1]$

然后莫反。
$f_x=\sum _{i=1}^n\sum _{j=1}^m\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{j}\rfloor [gcd(i,j)=x]$

$g_i=\sum _{x|i}{f}_i=\sum _{i=1}^n\sum _{j=1}^m\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{j}\rfloor [x|gcd(i,j)]$
$=\sum _{i=1}^n\sum _{j=1}^m\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{j}\rfloor \sum _{x|i,x|j}1$
$=\sum _x\sum _{i=1}^{\lfloor \frac{n}{x}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{x}\rfloor }\lfloor \frac{n}{ix}\rfloor \lfloor \frac{m}{jx}\rfloor$

$f_i=\sum _{i|x}\mu (\frac{x}{i})g_x$
$f_1=\sum _{1|x}\mu (\frac{x}{1})g_x=\sum _{i=1}^{\min (n,m)}\mu (i)g_i$

那询问我们可以预处理 $\mu$ 的前缀和然后用数论分块求 $g_i$。
（用数论分块预处理 $\sum _{i=1}^x\lfloor \frac{x}{i}\rfloor$）
（然后用数论分块搞 $\sum _x$ 的部分）

然后每次询问你就跑一次数论分块 $O(\sqrt{n})$，所以询问的复杂度是 $O(T\sqrt{n})$ 可以的。

代码

#include<cstdio>
#include<iostream>
#define ll long long

using namespace std;

int T, n, m, prime[50001];
ll f[50001], ans, miu[50001];
bool np[50001];

int main() {
	miu[1] = 1;
	for (int i = 2; i <= 50000; i++) {
		if (!np[i]) prime[++prime[0]] = i, miu[i] = -1;
		for (int j = 1; j <= prime[0] && i * prime[j] <= 50000; j++) {
			np[i * prime[j]] = 1;
			if (i % prime[j] == 0) {miu[i * prime[j]] = 0; break;}
				else miu[i * prime[j]] = -miu[i];
		}
	}
	for (int i = 1; i <= 50000; i++) miu[i] += miu[i - 1];
	
	for (int i = 1; i <= 50000; i++) {
		int r;
		for (int l = 1; l <= i; l = r + 1) {
			r = i / (i / l);
			f[i] += 1ll * (r - l + 1) * (i / l);
		}
	}
	
	scanf("%d", &T);
	while (T--) {
		scanf("%d %d", &n, &m);
		ans = 0;
		
		if (n > m) swap(n, m);
		
		int r;
		for (int l = 1; l <= n; l = r + 1) {
			r = min(n / (n / l), m / (m / l));
			ans += (miu[r] - miu[l - 1]) * f[n / l] * f[m / l];
		}
		
		printf("%lld\n", ans);
	}
	
	return 0;
}