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