Cod sursa(job #419284)

Utilizator vladiiIonescu Vlad vladii Data 17 martie 2010 11:32:14
Problema Pairs Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <iostream>
using namespace std;
#define LL long long
int N, M, NR[1000010], X[1000010];
bool A[1000010];
long long sol;

int main() {
    FILE *f1=fopen("pairs.in", "r"), *f2=fopen("pairs.out", "w");
    int as;
    long long i, j, p, q;
    fscanf(f1, "%d", &N);
    for(i=1; i<=N; i++) {
         fscanf(f1, "%d", &as);
         A[as]=1; M=max(M, as);
    }
    for(i=2; i<=M; i++) {
         for(j=i; j<=M; j+=i) {
              if(A[j]) { X[i]++; }
         }
    }
    for(i=2; i<=M; i++) {
         if(!NR[i]) {
              p=i*i;
              for(j=p; j<=M; j+=p) { NR[j]=-1; }
              for(j=i; j<=M; j+=i) {
                   if(NR[j]>-1) { NR[j]++; }
              }
         }
    }
    for(i=2; i<=M; i++) {
         if(NR[i]) {
              p=X[i]*(X[i]-1)/2;
              if(NR[i]%2) { 
                   sol+=p; 
              }
              else { sol-=p; }
         }
    }
    p=N*(N-1)/2; p=p-sol;
    fprintf(f2, "%lld\n", p);
    fclose(f1); fclose(f2);
    return 0;
}