Cod sursa(job #1968253)

Utilizator NicolaalexandraNicola Alexandra Mihaela Nicolaalexandra Data 17 aprilie 2017 16:19:27
Problema Pairs Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.68 kb
#include <fstream>

using namespace std;
int n,i,j,ap,nr,sol,d,p,e,maxim,v[100001],f[1000001],x[100001];
ifstream fin ("pairs.in");
ofstream fout ("pairs.out");

int main (){
// pentru fiecare numar i de la 1 la MAX verificam cate din numerele i, 2i, 3i ... [MAX/i]*i apartin multimii M. Consideram toate numerele P ce se scriu ca produs de numere prime, unde fiecare numar prim este folosit cel mult o data. Daca numarul de numere prime din descompunerea lui P este un numar impar, adunam xP * (xP-1) / 2 la Res. Altfel, din Res scadem xP * (xP-1) / 2.
    fin>>n;
    for (i=1;i<=n;i++){
        fin>>v[i];
        f[v[i]] = 1;
        maxim = max (maxim,v[i]);
    }
//    sort (v+1,v+n+1);
    for (i=2;i<=maxim;i++){
        ap = 0;
        for (j=i;j<=maxim;j+=i)
            if (f[j] == 1)
                ap++;
        x[i] = ap; // nr de nr care se divid cu i, din v
        // il descompunem pe i in factori primi
        d = 2;
        p = i;
        nr = 0;
        while (p!=1 && d*d<=p){
            e = 0;
            while (p%d == 0){
                p/=d;
                e++;
            }
            if (e == 1)
                nr++;
            if (e > 1){
                nr = -2;
                break;
            }
            d++;
        }
        if (p != 1)
            nr++;
        // nr = numarul de numere prime din descompunere care apar de maxim odata
        if (x[i] != 0 && nr > 0){
            if (nr%2 == 0){
                // scadem
                sol -= x[i]*(x[i]-1)/2;
            }
            else{
                sol += x[i]*(x[i]-1)/2;
            }
        }
    }
    fout<<n*(n-1)/2 - sol;


    return 0;
}