Cod sursa(job #1743950)

Utilizator LucianTLucian Trepteanu LucianT Data 18 august 2016 23:27:03
Problema Pairs Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <bits/stdc++.h>
#define maxN 1000005
using namespace std;
int p[maxN],ap[maxN];
int mobius[maxN];
int n,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]++;
    for(i=2;i<=maxN;i++)
        if(!p[i])
            for(j=i;j<=maxN;j+=i)
                p[j]=i;
    mobius[1]=1;
    for(i=2;i<=maxN;i++)
        if(p[i]!=p[i/p[i]])
            mobius[i]=(-1)*mobius[i/p[i]];
    for(i=1;i<=maxN;i++)
        if(mobius[i])
        {
            for(j=i+i;j<=maxN;j+=i)
                ap[i]+=ap[j];
            sol+=mobius[i]*(ap[i]*(ap[i]-1LL))/2;
        }
    printf("%lld",sol);
    return 0;
}