HDU 5072 Coprime（数论+容斥原理）

Description
给出n(3≤n≤10^5)个数字，每个数ai满足1≤ai≤10^5，求有多少对(a,b,c)满足[(a,b)=(b,c)=(a,c)=1] or [(a,b)≠1and(a,c)≠1and(b,c)≠1]，都互素或都不互素
Input
第一行一整数T表示用例组数，每组用例第一行为数字个数n，第二行n个互不相同的整数ai(T<=5,3<=n<=10^5,1<=ai<=10^5)
Output
对于每组用例，输出满足条件的(a,b,c)对数
Sample Input
1
5
1 3 9 10 2
Sample Output
4
Solution
假设a < b < c，总情况数C(n,3)，不合法情况如下
1.(a,b)!=1,(b,c)=1,(a,c)=1
2.(a,b)=1,(b,c)!=1,(a,c)=1
3.(a,b)=1,(b,c)=1,(a,c)!=1
4.(a,b)=1,(b,c)!=1,(a,c)!=1
5.(a,b)!=1,(b,c)=1,(a,c)!=1
6.(a,b)!=1,(b,c)!=1,(a,c)=1
记l1[i],l2[i],r1[i],r2[i]分别为比i小的与i互素的数的个数，比i小的与i不互素的数的个数，比i大的与i互素的数的个数，比i大的与i不互素的数的个数
则当i作为a时，考虑3和4以及1和6知不合法情况是r1[i]*r2[i]
当i作为b时，考虑1和4以及2和5知不合法情况是l1[i]*r2[i]+l2[i]*r1[i]
当i作为c时，考虑2和6以及3和5知不合法情况是l1[i]*l2[i]
故对于每个i，不合法情况就是(l1[i]+r1[i])*(l2[i]+r2[i])/2，即与其互素的数的数量乘上与其不互素的数的数量再除以二，而求与一个数互素的数的数量可以容斥求出
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 111111 
typedef long long ll;
int T,n,a[maxn];
int prime[maxn],res,is_prime[maxn],factor[11],num[maxn];
void get_prime()
{
    res=0;
    memset(is_prime,0,sizeof(is_prime));
    for(int i=2;i*i<maxn;i++)
        if(!is_prime[i])
        {
            prime[res++]=i;
            for(int j=2*i;j<maxn;j+=i)
                is_prime[j]=1;
        }
}
int get_factor(int x)
{
    int cnt=0;
    for(int i=0;i<res&&prime[i]<=x;i++)
        if(x%prime[i]==0)
        {
            factor[cnt++]=prime[i];
            while(x%prime[i]==0)
                x/=prime[i];
        }
    if(x>1)factor[cnt++]=x;
    return cnt;
}
int main()
{
    get_prime();
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        memset(num,0,sizeof(num));
        for(int k=1;k<=n;k++)
        {
            scanf("%d",&a[k]);
            int cnt=get_factor(a[k]);
            for(int i=0;i<(1<<cnt);i++)
            {
                int temp=1;
                for(int j=0;j<cnt;j++)
                    if(i&(1<<j))temp*=factor[j];
                num[temp]++;
            }
        }
        ll ans=0;
        for(int k=1;k<=n;k++)
        {
            int cnt=get_factor(a[k]),temp_ans=0;
            for(int i=0;i<(1<<cnt);i++)
            {
                int temp_cnt=0,temp=1;
                for(int j=0;j<cnt;j++)
                    if(i&(1<<j))
                        temp_cnt++,temp*=factor[j];
                if(temp_cnt&1)temp_ans-=num[temp];
                else temp_ans+=num[temp];
            }
            if(a[k]==1)continue;
            ans+=1ll*(temp_ans)*(n-1-temp_ans);
        }
        printf("%I64d\n",1ll*n*(n-1)*(n-2)/6-ans/2);
    }
    return 0;
}