hdoj 3980 Paint Chain（数学推导）

//以下为原blog搬迁过来的内容

【题目大意】：求两两互质和两两不互质的三个数的个数

【解题思路】：对第i个数分别求与它互质的和不互质的个数，记为p[i]，q[i]
sum+=p[i]*q[i]
ans=C(n,3)-sum/2

【代码】：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>
#include <string>
#include <cctype>
#include <map>
#include <iomanip>
                   
using namespace std;
                   
#define eps 1e-8
#define pi acos(-1.0)
#define inf 1<<30
#define pb push_back
#define lc(x) (x << 1)
#define rc(x) (x << 1 | 1)
#define lowbit(x) (x & (-x))
#define ll long long

int a[805],p[805],q[805];
int T,n,ans,summ;

int gcd(int a,int b)
{
    if(b==0)
        return a;
    else
        return gcd(b,a%b);
}

void init()
{
    memset(p,0,sizeof(p));
    memset(q,0,sizeof(q));
}

int main(){
    scanf("%d",&T);
    while(T--){
        summ=0;
        init();
        scanf("%d",&n);
        ans=n*(n-1)*(n-2)/6;
        for(int i=1;i<=n;i++)
           scanf("%d",&a[i]);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++) {
                if(i!=j) {
                    if(gcd(a[i],a[j])==1)
                        p[i]++;
                    else
                        q[i]++;
                }
            }
        for(int i=1;i<=n;i++)
            summ+=p[i]*q[i];
        ans=ans-summ/2;
        printf("%d\n",ans);
    }
    return 0;
}