分析:
求同色三角形的个数,我们用的是逆转换成求不同色三角形的个数,对于每个顶点,只要和它相连的两个边不同色,它就一定可以组成一个不同色三角形,而且一个三角形会被两次计算。所以一个顶点对不同色三角形的贡献是:
prime∗(n−1−unprime)/2
问题就转化成了求与一个数不互为质数的数的个数。可以对这个数质因数分解,然后容斥的减去所有含有这些质因数的数的个数,这是一个集合或上的问题显然要容斥。
所以这里预处理了一个数组
divide[i]
表示,能被
i
<script type="math/tex" id="MathJax-Element-62">i</script>整除的数的个数,来帮助容斥。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define pr(x) cout << #x << ": " << x << " "
#define pl(x) cout << #x << ": " << x << endl;
struct jibancanyang
{
static const int maxn = int(1e5) + 7;
int prime[maxn], factor[11];
bool isprime[maxn];
int divide[maxn], n, A[maxn], maxNum;
void getPrime() {
prime[0] = 0;
memset(isprime, 0, sizeof(isprime));
for (int i = 2; i < maxn; ++i) {
if (!isprime[i]) {
prime[++prime[0]] = i;
for (int j = i + i; j < maxn; j += i) isprime[j] = true;
}
}
}
void getFactor(int x) {
factor[0] = 0;
for (int i = 1; i <= prime[0] && prime[i] * prime[i] <= x; ++i) {
if (x % prime[i] == 0) {
factor[++factor[0]] = prime[i];
while (x % prime[i] == 0) x /= prime[i];
}
}
if (x > 1) factor[++factor[0]] = x;
}
void getDivide() {
for (int i = 2; i <= maxNum; ++i) {
for (int j = i + i; j <= maxNum; j += i) {
divide[i] += divide[j];
}
}
}
void fun() {
int T;
scanf("%d", &T);
while (T--) {
scanf("%d", &n);
memset(divide, 0, sizeof(divide));
maxNum = 0;
for (int i = 0; i < n; ++i) {
scanf("%d", &A[i]);
maxNum = max(A[i], maxNum);
divide[A[i]]++;
}
getDivide();
long long ans = (long long)(n) * (n - 1) * (n - 2) / 6;
long long unans = 0;
for (int i = 0; i < n; ++i) {
getFactor(A[i]);
long long temp = 0;
for (int s = 1; s < 1 << factor[0]; ++s) {
long long cur = 1;
int cnt = 0;
for (int j = 0; j < factor[0]; ++j) {
if (s >> j & 1) cur *= factor[j + 1], ++cnt;
}
if (cnt & 1) temp += divide[cur] - 1;
else temp -= divide[cur] - 1;
}
unans += temp * (n - 1 - temp);
}
printf("%lld\n", ans - unans / 2);
}
}
}ac;
int main()
{
#ifdef LOCAL
freopen("in.txt", "r", stdin);
//freopen("out.txt", "w", stdout);
#endif
ac.getPrime();
ac.fun();
return 0;
}