Cod sursa(job #2224324)

Utilizator a_h1926Heidelbacher Andrei a_h1926 Data 23 iulie 2018 18:33:19
Problema Principiul includerii si excluderii Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.1 kb
#include <fstream>
#include <vector>
#include <string>

using namespace std;

const string IN_FILE = "pinex.in";
const string OUT_FILE = "pinex.out";

vector<int64_t> factorize(const int64_t n) {
    auto factors = vector<int64_t>();
    int64_t x = n;
    for (int64_t f = 2; f * f <= x; f++) {
        if (x % f == 0) {
            factors.push_back(f);
            for (; x % f == 0; x/= f);
        }
    }
    if (x > 1) {
        factors.push_back(x);
    }
    return factors;
}

int64_t back(
        const int64_t n,
        const vector<int64_t>& factors,
        const int64_t p,
        const int i,
        const int sign) {
    if (i == int(factors.size())) return sign * n / p;
    return back(n, factors, p, i + 1, sign)
            + back(n, factors, factors[i] * p, i + 1, -sign);
}

int main() {
    ifstream in(IN_FILE);
    ofstream out(OUT_FILE);
    int Q;
    in >> Q;
    for (int i = 0; i < Q; i++) {
        int64_t a, b;
        in >> a >> b;
        const auto factors = factorize(b);
        out << back(a, factors, 1LL, 0, 1) << "\n";
    }
    in.close();
    out.close();
    return 0;
}