#include "souvenirs.h" #include <vector> #include <cstdint> #include <algorithm> void buy_souvenirs(int N, long long P0) { const long long M = P0 - 1; std::vector<int> bought(N, 0); for (int step = 0; step < 5000; step++) { bool done = true; for (int i = 1; i < N; i++) { if (bought[i] < i) { done = false; break; } } if (done) return; auto res = transaction(M); const std::vector<int>& L = res.first; for (int x : L) { if (x < N) bought[x]++; } } }