Cod sursa(job #3195154)

Utilizator amcbnCiobanu Andrei Mihai amcbn Data 20 ianuarie 2024 10:39:26
Problema Frac Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.63 kb
#include <bits/stdc++.h>
using namespace std;
const char nl = '\n';
const char sp = ' ';
const int inf = 0x3f3f3f3f;
const int mod = 1e9 + 9;
const char out[2][4]{ "NO", "YES" };
#define all(a) a.begin(), a.end()
using ll = long long;
ifstream fin("frac.in");
ofstream fout("frac.out");

const int pmax = 1e6;
bitset<pmax + 5> sieve;
vector<int> primes;
vector<ll> factors;
vector<ll> divisors;
ll n, p;

void precomputing() {
    sieve.set();
    for (int i = 2; i <= pmax; ++i) {
        if (sieve[i]) {
            primes.push_back(i);
            for (ll j = 1ll * i * i; j <= pmax; j += i) {
                sieve[j] = 0;
            }
        }
    }
}

ll check(ll a) {
    ll cnt = 0;
    for (auto& d : divisors) {
        cnt += a / d;
    }
    return cnt;
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    precomputing();
    fin >> n >> p;
    for (auto& p : primes) {
        if (1ll * p * p > n) {
            break;
        }
        if (n % p == 0) {
            factors.push_back(p);
            while (n % p == 0) {
                n /= p;
            }
        }
    }
    if (n > 1) {
        factors.push_back(n);
    }
    for (int mask = 0; mask < (1 << factors.size()); ++mask) {
        ll divisor = 1;
        for (int i = 0; i < mask; ++i) {
            if ((mask >> i) & 1) {
                divisor *= -factors[i];
            }
        }
        divisors.push_back(divisor);
    }
    ll answer = 0;
    for (ll i = 1ll << 61; i; i >>= 1) {
        if (check(answer + i) < p) {
            answer += i;
        }
    }
    fout << answer + 1;
}