hdu5072 coprime 容斥 和睦三元组

容斥的一个经典题型的变形。忘记ans清零怒WA七发。。

可以用改进的素数筛快速求出n个数中每个数的不互素数个数，效率大概是max(a[i])log(max(a[i]))。

虽然数据很水用暴力遍历每个数因数分解再进行容斥的方法也能过。。。

#include <stdio.h>#include <string.h>#include <algorithm>using namespace std;int num[100100],good[100100],cnt[100100],d[100100],n,m,a[100100];long long solve(){memset(good,1,sizeof(good));memset(cnt,0,sizeof(cnt));memset(d,0,sizeof(d));long long ans=0;for(int i=2;i<=a[n-1];i++){if(good[i]){if(d[i]==0){d[i]=1;}int nn=0;for(int j=1;i*j<=a[n-1];j++){if(num[i*j]){nn++;}}for(int j=1;i*j<=a[n-1];j++){if(d[i]==1&&j>1){if(j%i==0){good[i*j]=0;}else{++d[i*j];}}cnt[i*j]+=nn*(d[i]&1?1:-1);}}if(num[i]==1&&cnt[i]>0){ans+=(long long)(cnt[i]-1)*(n-cnt[i]);}}return (long long)n*(n-2)*(n-1)/6-(ans/2);}int main(){int t;scanf("%d",&t);while(t--){scanf("%d",&n);memset(num,0,sizeof(num));memset(nn,0,sizeof(nn));for(int i=0;i<n;i++){scanf("%d",&a[i]);num[a[i]]=1;}sort(a,a+n);printf("%lld\n",solve());}return 0;}