Cod sursa(job #1940154)

Utilizator Mircea_DonciuDonciu Mircea Mircea_Donciu Data 26 martie 2017 14:20:50
Problema Pairs Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <cstdio>
#include <algorithm>
using namespace std;
int n,m,x,i,j,v[1000005],p[1000005],q[1000005],a[1000005];
long long s;
int main()
{
    FILE *f=fopen("pairs.in","r");
    FILE *g=fopen("pairs.out","w");
    m=-1;
    fscanf(f,"%d",&n);
    for(i=1; i<=n; i++)
    {
        fscanf(f,"%d",&x);
        a[x]=1;
        if(m<x) m=x;
    }
    for(i=1; i<=m; i++)
        for(j=i; j<=m; j+=i)
            p[i]+=(a[j]==1);
    for(i=2; i<=m; i++)
    if(!v[i])
    {
            for(j=i; j<=m; j+=i)
                v[j]++;
            for(j=i*i; j<=m; j+=i*i)
                q[j]=1;
    }
    for(i=1; i<=m; ++i)
    {
        if(q[i]) continue;
        else
        {
            if(v[i]%2) s-=1LL*p[i]*(p[i]-1);
            else s+=1LL*p[i]*(p[i]-1);
        }
    }
    fprintf(g,"%lld\n",s/2);
    fclose(f); fclose(g);
    return 0;
}