Submission #491819

#	Time	Username	Problem	Language	Result	Execution time	Memory
491819		Victor	Brunhilda’s Birthday (BOI13_brunhilda)	C++17	100 / 100	411 ms	40036 KiB

brunhilda

// #pragma GCC target ("avx,avx2,fma")
// #pragma GCC optimize ("Ofast,inline") // O1 - O2 - O3 - Os - Ofast
// #pragma GCC optimize ("unroll-loops")
#include <bits/stdc++.h>

using namespace std;

#define rep(i, a, b) for (int i = (a); i < (b); ++i)
#define per(i, a, b) for (int i = (b - 1); i >= (a); --i)
#define trav(a, x) for (auto &a : x)

#define all(x) x.begin(), x.end()
#define sz(x) x.size()
#define pb push_back
#define debug(x) cout << #x << " = " << x << endl

#define umap unordered_map
#define uset unordered_set

typedef pair<int, int> ii;
typedef pair<int, ii> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<vi> vvi;

typedef long long ll;
typedef pair<ll, ll> pll;
typedef vector<ll> vll;
typedef vector<pll> vpll;

const int INF = 1'000'000'007;

int n, q, ans[10'000'001], rem;
priority_queue<ii> pq;

int main() {
    cin.tie(0)->sync_with_stdio(0);
    cin.exceptions(cin.failbit);

    int p, mx;
    ans[0] = 0;

    cin >> n >> q;
    rep(i, 0, n) {
        cin >> p;
        pq.emplace(0, p);
        mx = p;
    }

    int val, prime, stop = INF;
    rep(i, mx, 10'000'001) {
        while (1) {
            tie(val, prime) = pq.top();
            val = -val;

            if (i / prime == val / prime) {
                rem = i - val;
                if (!rem) {
                    ans[i] = INF;
                    stop = i;
                } else if (i - rem < mx)
                    ans[i] = 2;
                else 
                    ans[i] = ans[i - rem] + 1;
                break;

            } else {
                pq.pop();
                pq.emplace(-(i - i % prime), prime);
            }
        }
        if (stop != INF) break;
    }

    while (q--) {
        cin >> val;

        if (val<mx)
            cout<<1<<endl;
        else if (val>=stop)
            cout << "oo" << endl;
        else
            cout << ans[val] << endl;
    }
    return 0;
}

Compilation message (stderr)

brunhilda.cpp: In function 'int main()':
brunhilda.cpp:61:24: warning: 'mx' may be used uninitialized in this function [-Wmaybe-uninitialized]
   61 |                 } else if (i - rem < mx)
      |                        ^~

• Subtask 1: 20 / 20
• Subtask 2: 20 / 20
• Subtask 3: 60 / 60