Cod sursa(job #419296)

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

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