#include "souvenirs.h" #include <utility> #include <vector> #include <bits/stdc++.h> #define T std::pair<std::vector<int>, long long> using namespace std; void buy_souvenirs(int N, long long P0) { //n = 3; T res = transaction(P0-1ll); long long coin = P0 - 2ll - res.second; for (int tick = 2;tick < N;tick++) { for (int j = 0;j < tick;j++) res = transaction(coin); coin = coin - res.second - 1ll; if (res.first.size() == 2) { res = transaction(coin); coin = coin - res.second - 1ll; tick++; } } }