#include "souvenirs.h" #include <utility> #include <bits/stdc++.h> using namespace std; void buy_souvenirs(int N, long long P0) { pair<vector<int>, long long> res; if (N == 2) { transaction(P0 - 1); return; } else if (N == 3) { auto res1 = transaction(P0 - 1); long long x = res1.second; if (res1.first[2] == 0) { transaction(P0 - x - 2); transaction(P0 - x - 2); } else { transaction((P0 - x - 1)/2); } return; } }