hdu5072（鞍山regional problem C）：容斥，同色三角形模型

#include <iostream>
#include <stdio.h>
#include<string.h>
#include<algorithm>
#include<string>
#include<ctype.h>
using namespace std;
#define MAXN 1000
long long n;
int a[100010];
int prime[100010];
int num[100010];
int isnotprime[100010];
int num_prime=0;
int fac[100010][10];
int np[100010];
long long gcd(long long a,long long b)
{
    return b?gcd(b,a%b):a;
}
long long lcm(long long a,long long b)
{
    return a/gcd(a,b)*b;
}
void setprime()
{
    for(int i = 2 ; i < MAXN ; i ++)
    {
        if(!isnotprime[i])
             prime[num_prime ++]=i;
        for(int j=0;j<num_prime&&i*prime[j]<MAXN;j++)
        {
            isnotprime[i * prime[j]] = 1;
            if(!(i%prime[j] ) )
                break;
        }
    }
    return ;
}
void setfac(int x,int pos)
{
    np[pos]=0;
    for(int i=0;i<num_prime;i++)
    {
        if(!(x%prime[i]))
        {
            fac[pos][np[pos]++]=prime[i];
        }
        while(!(x%prime[i]))
        {
            x/=prime[i];
        }
    }
    if(x>1)
    {
        fac[pos][np[pos]++]=x;
    }
}
long long iae(int pos)
{
    long long res=0;
    for(int i=1;i<(1<<np[pos]);i++)
    {
        long long mut=1,tmp=0;
        for(int j=0;j<np[pos];j++)
        {
            if(i&(1<<j))
            {
                mut*=fac[pos][j];
                tmp++;
            }
        }
        if(tmp&1)
        {
            res+=num[mut]-1;
        }
        else
        {
            res-=num[mut]-1;
        }
    }
    return res;
}
void setnum()
{
    for(int i=2;i<=100000;i++)
    {
        for(int j=i+i;j<=100000;j+=i)
            num[i]+=num[j];
    }
}
int main()
{
    #ifndef ONLINE_JUDGE
        //freopen("in.txt","r",stdin);
    #endif
    int T;
    setprime();
    scanf("%d",&T);
    while(T--)
    {
        memset(num,0,sizeof(num));
        scanf("%I64d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d",a+i);
            num[a[i]]++;
            setfac(a[i],i);
        }
        setnum();
        long long ans=0;
        for(int i=0;i<n;i++)
        {
            int tmp=iae(i);
            ans+=(n-1-tmp)*tmp;
        }
        ans=n*(n-1)*(n-2)/6-ans/2;
        printf("%I64d\n",ans);
    }
    return 0;
}