洛谷P3911 最小公倍数之和 莫比乌斯反演经典转换

题目链接

题解：

刚拿到这道题目，发现是$n$个离散的数，很难处理。因此，我们需要对问题进行转化：

$$\sum _{i=1}^N\sum _{j=1}^{N}lcm(i,j)\ast cnt[i]\ast cnt[j]$$

$cnt$表示该数出现的次数。

下面可以对上面的式子进行化简：

${\sum }_{i=1}^N{\sum }_{j=1}^{N}lcm(i,j)\ast cnt[i]\ast cnt[j]$

$={\sum }_{i=1}^N{\sum }_{j=1}^N\frac{i\ast j}{gcd(i,j)}\ast cnt[i]\ast cnt[j]$

$={\sum }_{k=1}^N{\sum }_{i=1}^N{\sum }_{j=1}^N[gcd(i,j)==k]\frac{i\ast j}{k}\ast cnt[i]\ast cnt[j]$

$={\sum }_{k=1}^N{\sum }_{i=1}^{\frac{N}{k}}{\sum }_{j=1}^{\frac{N}{k}}[gcd(i,j)==1]i\ast j\ast k\ast cnt[ik]\ast cnt[jk]$

$={\sum }_{k=1}^N{\sum }_{i=1}^{\frac{N}{k}}{\sum }_{j=1}^{\frac{N}{k}}{\sum }_{d|gcd(i,j)}\mu (d)\ast i\ast j\ast k\ast cnt[ik]\ast cnt[jk]$

$={\sum }_{k=1}^N{\sum }_{d=1}^{\frac{N}{k}}{\sum }_{i=1}^{\frac{N}{kd}}{\sum }_{j=1}^{\frac{N}{kd}}{d}^2\ast \mu (d)\ast i\ast j\ast k\ast cnt[ikd]\ast cnt[jkd]$

$={\sum }_{k=1}^N{\sum }_{kd=1}^N{d}^2\ast \mu (d){\sum }_{i=1}^{\frac{N}{kd}}{\sum }_{j=1}^{\frac{N}{kd}}i\ast j\ast k\ast cnt[ikd]\ast cnt[jkd]$

$={\sum }_{T=1}^{N}T({\sum }_{d|T}d\ast \mu (d)){\sum }_{i=1}^{\frac{N}{T}}{\sum }_{j=1}^{\frac{N}{T}}i\ast j\ast cnt[iT]\ast cnt[jT]$ 注意这步经典转换

$={\sum }_{T=1}^{N}T({\sum }_{d|T}d\ast \mu (d))({\sum }_{i=1}^{\frac{N}{T}}cnt[i\ast T]\ast i{)}^2$

发现这个式子现在不能进一步进行优化，因此，只能转化到这个式子。

下面计算这个式子。

我们可以对第一个括号内的式子进行预处理，第二个括号内的式子可以暴力求解。整个程序的时间复杂度就是$O(nlogn)$。这题数据量为$n\le 50000$，可以通过。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int MAXN = 1e6 + 10;
const int mod = 1e9 + 7;
int primes[MAXN], cnt, mu[MAXN], f[MAXN], cnt1[MAXN];
bool vis[MAXN];
void init(int n) {
	mu[1] = 1;
	for (int i = 2; i <= n; i++) {
		if (!vis[i]) {
			primes[cnt++] = i;
			mu[i] = -1;
		}
		for (int j = 0; primes[j] <= n / i; j++) {
			vis[i * primes[j]] = true;
			if (i % primes[j] == 0) {
				mu[i * primes[j]] = 0;
				break;
			}
			mu[i * primes[j]] = -mu[i];
		}
	}
	for (int i = 1; i <= n; i++) {
		for (int j = i; j <= n; j += i) {
			f[j] += mu[i] * i;
		}
	}
}
signed main() 
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int n;
	init(50005);
	cin >> n;
	for (int i = 0; i < n; i++) {
		int x;
		cin >> x;
		cnt1[x]++;
	}
	int ans = 0;
	for (int i = 1; i <= 50000; i++) {
		int now = 0;
		for (int j = 1; j <= 50000 / i; j++) {
			now += cnt1[j * i] * j;
		}
		ans += i * f[i] * now * now;
	}
	cout << ans << endl;
    return 0;
}