Cod sursa(job #3173493)

Utilizator Radu_MocanasuMocanasu Radu Radu_Mocanasu Data 22 noiembrie 2023 22:33:15
Problema Pairs Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.77 kb
#pragma GCC optimize("Ofast,inline,unroll-loops")
#include <bits/stdc++.h>
#define NMAX  1000000
using namespace std;
ifstream fin("pairs.in");
ofstream fout("pairs.out");
bitset <NMAX + 5> c;
int fp[NMAX + 5];
int mp[NMAX + 5];
int t[NMAX + 5];
void ciur(){
    c[2] = 1;
    for(int i = 2; i <= NMAX; i += 2){
        c[i] = 1;
        fp[i] = 2;
    }
    for(int i = 3; i * 2 <= NMAX; i++){
        if(!c[i]){
            for(int j = i; j <= NMAX; j += i){
                fp[j] = i;
                c[j] = 1;
            }
        }
    }
}
int nrdp(int n){
    int s = 0;
    while(n > 1){
        int k = 0,f = fp[n];
        if(f == 0) f = n;
        while(n % f == 0){
            n /= f;
            k++;
        }
        if(k > 1) return 0;
        s++;
    }
    return s;
}

int main()
{
    int n,i,d,x;
    ciur();
    fin >> n;
    for(i = 1; i <= n; i++){
        fin >> x;
        for(d = 2; d * d < x; d++){
            if(x % d == 0){
                if(t[d] || nrdp(d)){
                    mp[d]++;
                    if(!t[d]) t[d] = nrdp(d);
                }
                if(t[x / d] || nrdp(x / d)){
                    mp[x / d]++;
                    if(!t[x / d]) t[x / d] = nrdp(x / d);
                }
            }
        }
        if(d * d == x && (t[d] || nrdp(d))){
            mp[d]++;
            if(!t[d]) t[d] = nrdp(d);
        }
        if(t[x] || nrdp(x)){
            mp[x]++;
            if(!t[x]) t[x] = nrdp(x);
        }
    }
    long long rez = 1LL * n * (n - 1) / 2;
    for(i = 2; i <= NMAX; i++){
        if(!mp[i]) continue;
        if(t[i] % 2 == 1) rez -= 1LL * mp[i] * (mp[i] - 1) / 2;
        else rez += 1LL * mp[i] * (mp[i] - 1) / 2;
    }
    fout << rez;
    return 0;
}