题意:给出n个数,要你选出三个数, a,b,c 两两之间要么互质, 要么都不互质 求有多少个这样的三元组
思路:直接求肯定不行,但我们考虑反面就只有一种情况,选出的三元组存在一组互质一组不互质 那么单色三角形模型+容斥做咯,但是在前面还要预处理能被某个因子整除的数的个数
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<stack>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
const int qq = 1e5+5;
ll cnt[qq], x[qq];
int n;
int main(){
int t;scanf("%d",&t);
while(t--){
memset(cnt, 0, sizeof(cnt));
memset(x, 0 ,sizeof(x));
scanf("%d", &n);
ll a;
for(int i=0; i<n; ++i){
scanf("%lld", &a);
x[a]++;
}
for(int i=1; i<qq; ++i) //求能被1到1e5整除因子的个数、
for(int j=i; j<qq; j+=i)
cnt[i]+=x[j];
ll ans=0;
vector<int>v;
for(int i=1; i<qq; ++i){ //容斥原理求结果
if(x[i]){ //i这个数和其他数组成的三元组其实对于和i相等的数也成立,但是我们要找不同的三元组,
v.clear(); //所以只需要算其中一个数的贡献、
int t = i;
for(int j=2; j*j<qq&&t>1; ++j) //分解质因数、
if(t%j==0){
v.push_back(j);
while(t%j==0) t/=j;
}
if(t>1) v.push_back(t);
if(v.size()==0) continue;
int num = v.size();
ll res=0;
for(int j=1; j<(1<<num); ++j){
int p=0; ll k=1;
for(int l=0; l<num; ++l)
if(j&(1<<l)) p++, k*=v[l];
if(p%2==1) res+=cnt[k]; //这里实际上是求所有数中能被k整除的个数
else res-=cnt[k]; //普通的容斥是 用n/k(n代表数范围1~n)就是求1~n中能被k整除的个数、此处理由一样
}
ans+=(ll)max(0LL, res-1)*(n-res); //res-1 每次算其实都包含了自己 所以要减去
}
}
printf("%lld\n", (ll)n*(n-1)*(n-2)/6-ans/2);
}
return 0;
}