Cod sursa(job #1419657)

Utilizator chiriacandrei25Chiriac Andrei chiriacandrei25 Data 16 aprilie 2015 01:14:29
Problema Pairs Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.46 kb
#include <cstdio>
#define ValMax 1000005

using namespace std;

bool fr[ValMax];
int cnt[ValMax],a[100005],len[100005],LEN;
int v[100005][9];

inline void Desc(int lin, int x)
{
    int d=3,k=0;
    len[lin]=0;
    while(x%2==0)
    {
        ++k;
        x/=2;
    }
    if(k) v[lin][++len[lin]]=2;
    while(x>1 && d*d<=x)
    {
        k=0;
        while(x%d==0)
        {
            ++k;
            x/=d;
        }
        if(k) v[lin][++len[lin]]=d;
        d+=2;
    }
    if(x>1) v[lin][++len[lin]]=x;
}

inline void Back(int pas, int produs)
{
    if(pas==len[LEN]+1)
    {
        ++cnt[produs];
        return;
    }
    Back(pas+1,produs);
    Back(pas+1,produs*v[LEN][pas]);
}

int main()
{
    int n,i,j,x,nr,produs,stare;
    long long sol=0;
    freopen ("pairs.in","r",stdin);
    freopen ("pairs.out","w",stdout);
    scanf("%d", &n);
    for(i=1;i<=n;++i)
    {
        scanf("%d", &a[i]);
        Desc(i,a[i]);
        LEN=i;
        Back(1,1);
    }
    cnt[1]=n;
    for(i=1;i<=n;++i)
    {
        for(stare=1;stare<(1<<len[i]);++stare)
        {
            produs=1;
            for(j=nr=0;j<len[i];++j)
                if((1<<j)&stare)
                {
                    ++nr;
                    produs=produs*v[i][j+1];
                }
            if(nr&1) sol+=cnt[produs]-1;
            else sol-=cnt[produs]-1;
        }
    }
    printf("%lld\n", 1LL*n*(n-1)/2-sol/2);
    return 0;
}