LA 7048 Coprime 莫比乌斯反演

题意：

给出\(n(n \leq 10^5)\)个数字\(a_i(a_i \leq 10^5)\)，从中选出\(3\)个数，使得这\(3\)个数两两互质或者两两不互质

分析：

可以说这是《训练指南》\(P_{105}\)上问题\(6\)的原题。

将\(n\)个数看成\(n\)个顶点，如果两数互质连一条白边，不互质连一条黑边。

那么我们要计数的就是单色三角形的个数。

从\(n\)个数中选\(3\)个数，一共有\(C_n^3\)种方案，正面不容易计算所以我们反面计算非单色三角形的个数。

在一个非单色三角形中，恰好有两个顶点连接两条异色边。

而且有公共顶点的两条异色边对应一个非单色三角形。

假设与\(a_i\)互质的数字的个数为\(b_i\)(相当于连了\(b_i\)条白边)，那么与\(a_i\)不互质的数字的个数为\(n-1-b_i\)(连了\(n-1-b_i\)条黑边)

每个非单色三角形被计算了两次，所以对应的个数为$ \frac {1} {2} \sum{b_i (n-1-b_i)}$

最后单色三角形的个数就是\(C_n^3\)减去非单色三角形的个数。

关于计算与\(a_i\)互质的数字的个数，根据莫比乌斯反演公式有 $ \sum{\mu(d) cnt_d, (d | a_i)} \(，其中\)cnt_d\(为\)d$的倍数的个数。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;

typedef long long LL;
const int maxn = 100000;

int mu[maxn + 10], pcnt, prime[maxn];
bool vis[maxn + 10];
vector<int> factors[maxn + 10];

void preprocess() {
pcnt = 0;
mu[1] = 1;
for(int i = 2; i <= maxn; i++) {
if(!vis[i]) {
mu[i] = -1;
prime[pcnt++] = i;
}
for(int j = 0; j < pcnt && i * prime[j] <= maxn; j++) {
vis[i * prime[j]] = true;
if(i % prime[j] != 0) mu[i * prime[j]] = -mu[i];
else {
mu[i * prime[j]] = 0;
break;
}
}
}

for(int i = 2; i <= maxn; i++) if(mu[i])
for(int j = i; j <= maxn; j += i) factors[j].push_back(i);
}

int n, a[maxn + 10], cnt[maxn + 10];

int main()
{
preprocess();

int T; scanf("%d", &T);
while(T--) {
scanf("%d", &n);

memset(cnt, 0, sizeof(cnt));
for(int i = 0; i < n; i++) {
scanf("%d", a + i);
for(int d : factors[a[i]]) cnt[d]++;
}

LL ans = 0;
for(int i = 0; i < n; i++) {
LL coprime = n;
for(int d : factors[a[i]]) coprime += mu[d] * cnt[d];
if(a[i] == 1) coprime--;
ans += coprime * (n - 1 - coprime);
}
ans >>= 1;
LL tot = (LL)n * (n-1) * (n-2) / 6;
printf("%lld\n", tot - ans);
}

return 0;
}