[51Nod - 1594]Gcd and Phi（莫比乌斯反演）

题面

求 $\sum_{i=1}^n\sum_{j=1}^n\varphi\left(gcd(\varphi(i),\varphi(j))\right)$
$n\le 2e6$

题解

先初步推一推柿子。
$Ans=\sum_{d=1}^n\varphi(d)\sum_{i=1}^n\sum_{j=1}^n[gcd(\varphi(i),\varphi(j))==d]$用 $s(k)$表示 $n$以内 $\varphi(i)=k$的正整数 $i$的个数。则
$Ans=\sum_{d=1}^n\varphi(d)\sum_{i=1}^n\sum_{j=1}^ns(i)\cdot s(j)\cdot[gcd(i,j)==d]\\=\sum_{d=1}^n\varphi(d)\sum_{i=1}^{\lfloor\frac nd\rfloor}\sum_{j=1}^{\lfloor\frac nd\rfloor}s(id)\cdot s(jd)\cdot[gcd(i,j)==1]$令 $g(d)=\sum_{i=1}^{\lfloor\frac nd\rfloor}\sum_{j=1}^{\lfloor\frac nd\rfloor}s(id)\cdot s(jd)\cdot[gcd(i,j)==1]$，发现不好求。

再令 $f(d)=\sum_{i=1}^{\lfloor\frac nd\rfloor}\sum_{j=1}^{\lfloor\frac nd\rfloor}s(id)\cdot s(jd)$，有：
$f(d)=\sum_{k=1}^{\lfloor\frac nd\rfloor}g(kd)$反演后得：
$g(d)=\sum_{k=1}^{\lfloor\frac nd\rfloor}\mu(k)f(kd)\\ \therefore Ans=\sum_{i=1}^n\varphi(n)g(\lfloor\frac nd\rfloor)=\sum_{d=1}^n\varphi(d)\sum_{i=1}\mu(i)f(id)$

那么预处理出来 $f$，就能求 $g$了。直接按照式子 $O(n\log n)$求。就完事了。

$O(Tn\log n)$，有点慢还是能过。还有个小优化可以把 $\varphi$和 $\mu$枚举顺序交换一下，那么对于 $\mu=0$的时候就不用算（也优化不了多少）。

CODE

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2000005;
int n, cnt, p[N], phi[N], s[N], mu[N];
bool vis[N];
LL f[N];

void init(int N) {
	phi[1] = mu[1] = 1;
	for(int i = 2; i <= N; ++i) {
		if(!vis[i]) p[++cnt] = i, phi[i] = i-1, mu[i] = -1;
		for(int j = 1, k; j <= cnt && i * p[j] <= N; ++j) {
			vis[k = i * p[j]] = 1;
			if(i % p[j] == 0) {
				mu[k] = 0;
				phi[k] = phi[i] * p[j];
				break;
			}
			mu[k] = -mu[i];
			phi[k] = phi[i] * (p[j]-1);
		}
	}
}

int main () {
	init(2000000); int T, n;
	scanf("%d", &T);
	while(T--) {
		scanf("%d", &n);
		for(int i = 1; i <= n; ++i) ++s[phi[i]];
		for(int i = 1; i <= n; ++i) {
			f[i] = 0;
			for(int j = i; j <= n; j += i)
				f[i] += s[j];
			f[i] = f[i] * f[i];
		}
		LL ans = 0;
		for(int i = 1; i <= n; ++i) if(mu[i]) {
			LL sum = 0;
			for(int d = 1; i*d <= n; ++d)
				sum += phi[d] * f[i*d];
			ans += sum * mu[i];
		}
		printf("%lld\n", ans);
		for(int i = 1; i <= n; ++i) --s[phi[i]];
	}
}