Cod sursa(job #2519851)

Utilizator IulianOleniucIulian Oleniuc IulianOleniuc Data 8 ianuarie 2020 15:02:05
Problema Pairs Scor 0
Compilator cpp-64 Status done
Runda RoadToIOI #6 Marime 1.35 kb
#include <bits/stdc++.h>
using namespace std;

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

int main() {
    int n; fin >> n;
    vector<int> prm(n + 1);
    vector<int> ind(1e6 + 1);
    for (int i = 1; i <= n; i++) {
        fin >> prm[i];
        ind[prm[i]] = i;
    }

    vector<bool> sieve(1e6 + 1);
    vector<vector<int>> div(n + 1);
    for (int i = 2; i <= 1e6; i++)
        if (!sieve[i]) {
            if (ind[i])
                div[ind[i]].push_back(i);
            for (int j = 2 * i; j <= 1e6; j += i) {
                sieve[j] = true;
                if (ind[j])
                    div[ind[j]].push_back(i);
            }
        }

    vector<int> frq(1e6 + 1);
    for (int i = 1; i <= n; i++)
        for (int mask = 1; mask < (1 << div[i].size()); mask++) {
            int prd = 1, cnt = 0;
            for (int bit = 0; bit < (int) div[i].size(); bit++)
                if (mask & (1 << bit)) {
                    prd *= div[i][bit];
                    cnt++;
                }
            frq[prd] += (cnt % 2 ? +1 : -1);
        }

    vector<int> dp(1e6 + 1, n - ind[1]);
    for (int i = 2; i <= 1e5; i++)
        for (int j = i; j <= 1e5; j += i)
            dp[j] -= frq[i];

    int64_t sol = 0;
    for (int i = 1; i <= n; i++)
        sol += dp[prm[i]];
    fout << sol / 2 + ind[1] * n << '\n';

    fout.close();
    return 0;
}