题解 P4466 [国家集训队]和与积

题目描述

给出$N$，统计满足下面条件的数对$(a,b)$的个数：

1. $1\le a<b\le N$
2. $a+b$整除$a\times b$

输入输出格式

输入格式

一行一个数$N$

输出格式

一行一个数表示答案

思路讲解

比较好的莫比乌斯反演题了。

$$\text{ans}=\sum _{a=1}^N\sum _{b=a+1}^N[(a+b)|ab]$$
考虑到令：$d=(a,b),a=i\cdot d,b=j\cdot d$
$$(id+jd)|ij{d}^2\iff d(i+j)|ij{d}^2\iff (i+j)|ijd$$
又有$(i+j,ij)=1$，所以只能有：
$$(i+j)|d$$
给出合法的$(i,j)$，求出满足条件的$d$的个数。

$$=\sum _{i=1}^{\sqrt{N}}\sum _{j=i+1}^{\sqrt{N}}[(i,j)=1]\lfloor \frac{n}{j(i+j)}\rfloor$$
不难想到反演：
$$=\sum _{i=1}^{\sqrt{N}}\sum _{j=1}^{\sqrt{N}}\sum _{\delta |(i,j)}\mu (\delta )\lfloor \frac{n}{j(i+j)}\rfloor$$
将$\mu (\delta )$的和式提前：
$$=\sum _{\delta =1}^{\sqrt{N}}\mu (\delta )\sum _{i=1}^{\lfloor \frac{\sqrt{N}}{\delta }\rfloor }\sum _{j=i+1}^{\lfloor \frac{\sqrt{N}}{\delta }\rfloor }\lfloor \frac{N}{{\delta }^{2}j(i+j)}\rfloor$$
$$=\sum _{\delta =1}^{\sqrt{N}}\mu (\delta )\sum _{i=1}^{\lfloor \frac{\sqrt{N}}{\delta }\rfloor }\sum _{j=i+1}^{\lfloor \frac{\sqrt{N}}{\delta }\rfloor }\lfloor \frac{\lfloor \frac{N}{j{\delta }^2}\rfloor }{i+j}\rfloor$$
这个算法是正确的，但仍需要优化。

可以令$\alpha =i+j$:
$$=\sum _{\delta =1}^{\sqrt{N}}\mu (\delta )\sum _{j=1}^{\lfloor \frac{\sqrt{N}}{\delta }\rfloor }\sum _{\alpha =j+1}^{2j-1}\lfloor \frac{\lfloor \frac{N}{j{\delta }^2}\rfloor }{\alpha }\rfloor$$
先预处理莫比乌斯函数，后面就是个分块了。

#include <cstdio>
#include <iostream>
#include <cmath>

using namespace std;
const int N = 1e+5 + 1;
typedef long long ll;
int n, m, tot, Mu[N], P[N];
bool flag[N];
ll ans;

void sieve() {
	Mu[1] = 1;
	for(int i = 2; i <= m; i++) {
		if(!flag[i]) P[++tot] = i, Mu[i] = -1;
		for(int j = 1; j <= tot; j++) {
			if(i * P[j] > m) break;
			flag[i * P[j]] = 1;
			Mu[i * P[j]] = -Mu[i];
			if(i % P[j] == 0) {Mu[i * P[j]] = 0; break;}
		}
	}
}
ll L(int x, int y) {
	ll res = 0; int z = x << 1;
	if(!y) return 0;
	for(int l = x + 1, r; l < z; l = r + 1) {
		if(!(y / l)) return res;
		r = min(y / (y / l), z - 1);
		res += (r - l + 1) * (y / l);
	}
	return res;
}
int main()
{
	scanf("%d", &n);
	m = sqrt(n);
	sieve();
	for(int k = 1; k <= m; k++) {
		if(!Mu[k]) continue;
		for(int x = 1; x * k <= m; x++)
			ans += Mu[k] * L(x, n / (x * k * k));
	}
	printf("%lld", ans);
	return 0;
}