Cod sursa(job #1743945)

Utilizator LucianTLucian Trepteanu LucianT Data 18 august 2016 23:24:27
Problema Pairs Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <bits/stdc++.h>
#define maxN 1000005
using namespace std;
int p[maxN],ap[maxN];
int mobius[maxN];
int n,maxx,i,j,x,val;
long long sol;
bool seen[maxN];
int main()
{
    freopen("pairs.in","r",stdin);
    freopen("pairs.out","w",stdout);
    scanf("%d",&n);
    for(i=1;i<=n;i++)
        scanf("%d",&j),
        ap[j]++,
        maxx=max(maxx,j);
    for(i=2;i<=maxx;i++)
        if(!p[i])
            for(j=i;j<=maxx;j+=i)
                p[j]=i;
    mobius[1]=1;
    for(i=2;i<=maxx;i++)
        if(p[i]!=p[i/p[i]])
            if(mobius[i/p[i]]==1)
                mobius[i]=-1;
            else mobius[i]=1;
    for(i=1;i<=maxx;i++)
        if(mobius[i])
        {
            for(j=i+i;j<=maxx;j+=i)
                ap[i]+=ap[j];
            sol+=mobius[i]*(ap[i]*(ap[i]-1))/2;
        }
    printf("%lld",sol);
    return 0;
}