Cod sursa(job #3279151)

Utilizator paull122Paul Ion paull122 Data 21 februarie 2025 22:53:12
Problema Pairs Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <bits/stdc++.h>

#define NMAX 1000000
#define LOG 20

#define ll long long int
#define BASE 128
#define MOD 1000000007


using namespace std;

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

int mobius[NMAX+1];
vector<int> divs[NMAX+1];
void pre()
{
    mobius[1]=1;
    for(int i=1;i<=NMAX;i++)
    {
        divs[i].push_back(i);
        for(int j=2;j*1ll*i<=NMAX;j++)
        {
            mobius[i*j] -= mobius[i];
            divs[i*j].push_back(i);
        }
    }
}
ll cnt[NMAX+1];
int main()
{
    pre();
    int n;
    fin >> n;
    for(int i=1;i<=n;i++)
    {
        int x;
        fin >> x;
        for(int d : divs[x])
        {
            cnt[d]++;
        }
    }
    ll res=0;
    for(int i=1;i<=NMAX;i++)
    {
        res += mobius[i]*cnt[i]*(cnt[i]-1)/2;
    }
    fout << res;
}