#include "souvenirs.h" #include <utility> #include <vector> #include <stdint.h> #include <algorithm> void buy_souvenirs(int N, long long P0) { const int32_t M = P0 - 1; std::vector<int32_t> bought(N, 0); for(int32_t typeIdx = 1; typeIdx < N; typeIdx++) { while(bought[typeIdx] <= typeIdx) { auto transactionResult = transaction(M); const auto& souvenirs = transactionResult.first; for(uint32_t i = 0; i < souvenirs.size(); i++) { bought[souvenirs[i]]++; } } } return; }