Cod sursa(job #2307261)

Utilizator mihaicivMihai Vlad mihaiciv Data 24 decembrie 2018 09:55:44
Problema Pairs Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.43 kb
#include <iostream>
#include <fstream>
using namespace std;
const int NMAX = 1000010;
int vis[NMAX + 1];
long long int n;
int x[NMAX + 1];
int euler[NMAX + 1];
long long int answer;
long long int vmax;

void generateEuler() {
    // euler[i] = -1 => are divizori primi cu exponent > 1
    // euler[i] = v > 0 => nr de divizori primi ai lui i este v
    for (int i = 2; i <= vmax; ++i) {
        if (euler[i] == 0) {
            for (int j = i; j <= vmax; j = j + i) {
                if ( (j / i ) % i == 0 ) {
                    euler[j] = -1;
                } else if (euler[j] != -1) {
                    euler[j] ++;
                }
            }
        }
        if (euler[i] != -1 && euler[i] != 0) {
            if (euler[i] % 2 == 0 ) {
                answer = answer - (x[i] * (x[i] - 1) / 2);
            } else {
                answer = answer + (x[i] * (x[i] - 1) / 2);
            }
        }
    }

}

int main() {

    ifstream f("pairs.in");
    ofstream g("pairs.out");

    f >> n;
    for (int i = 0; i < n; ++i) {
        long long int x;
        f >> x;
        vis[x] = 1;
        vmax = max(vmax, x);
    }

    for (int i = 2; i <= vmax; ++i) {
        for (int j = i; j <= vmax; j += i) {
            if (vis[j] != 0) {
                x[i] ++;
            }
        }
    }
    generateEuler();
    //cout << euler[20];
    g << n * (n - 1) / 2 - answer;


    return 0;
}