Cod sursa(job #2506797)

Utilizator victoreVictor Popa victore Data 8 decembrie 2019 19:43:52
Problema Pairs Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.4 kb
//check check check
#include<iostream>
#include<vector>
#include<algorithm>
#include<fstream>
#include<queue>
#include<cstring>
#include<map>
#include<iomanip>
#include<set>
#include<unordered_map>

#define ll long long
#define pb(x) push_back(x)

using namespace std;

typedef pair<int,int> ii;

const int NMAX = 1e5+5;
const int VALMAX = 1e6+5;

ifstream fin("pairs.in");
ofstream fout("pairs.out");

bool mp[VALMAX];
int viz[VALMAX],cntdiv[VALMAX];

int main()
{
    ll i,j;
    int maxim = 1;
    ll N;
    fin>>N;
    ll rez = N*(N-1)/2;
    for(i = 1 ; i <= N ; ++i)
    {
        int x;
        fin>>x;
        mp[x]++;
        maxim = max(maxim , x);
    }

    //cnt[i] = cate numere din M sunt divizibile cu i
    for(i = 2 ; i <= maxim ; ++i)
    {
        if(cntdiv[i] == 0)
        {
            for(j = i ; j <= maxim ; j+= i)
                cntdiv[j]++;

            for(j = i*i ; j <= maxim ; j += i*i)
                viz[j] = 1;
        }

        if(viz[i] == 0)
        {
            ll cnt = 0;
            for(j = i ; j <= maxim ; j += i)
            {
                cnt += mp[j];
            }
            if(cntdiv[i]&1)
            {
                rez -= ((cnt*(cnt-1))>>1);
            }
            else
            {
                rez += ((cnt*(cnt-1))>>1);
            }
        }
    }

    fout<<rez;

    return 0;
}