Cod sursa(job #2657026)

Utilizator PredescuSebastianIonPredescu Sebastian Ion PredescuSebastianIon Data 9 octombrie 2020 15:29:37
Problema Pairs Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <fstream>

using namespace std;
ifstream f("pairs.in");
ofstream g("pairs.out");
int m,Max,ok[1000005],nr[1000005];
long long n,k,s;
bool marker[1000005],prim[1000005];
int main()
{
    f>>n;
    for(int i=1; i<=n; i++)
    {
        f>>m;
        marker[m]=1;
        Max=max(Max,m);
    }
    for(int i=2; i<=1000000; i++)
    {
        if(!prim[i])
        {
            nr[i]=1;
            ok[i]=0;
            for(int j=2*i; j<=1000000; j+=i)
            {
                prim[j]=1;
                nr[j]++;
                if((j/i)%i==0)ok[j]=1;
            }
        }
    }
    for(int i=2; i<=Max; i++)
    {
        if(ok[i])continue;
        k=0;
        for(int j=i; j<=Max; j+=i)
        {
            k+=marker[j];
        }
        if(nr[i]%2==1)s+=1LL*k*(k-1)/2;
        else s-=1LL*k*(k-1)/2;
    }
    g<<1LL*n*(n-1)/2-s<<'\n';
    return 0;
}