#include "souvenirs.h" #include <bits/stdc++.h> using namespace std; #define int long long #define pb push_back #define mp make_pair #define pii pair<int, int> #define fi first #define se second #define typ pair<vector<signed>, int> void solve_2(int n, int p0) { transaction(p0 - 1); } void solve_3(int n, int p0) { typ q1 = transaction(p0 - 1); if(q1.fi.size() == 1) { int p1 = (p0 - 1) - q1.se; transaction(p1 - 1); transaction(p1 - 1); } else { int p1_plus_p2 = (p0 - 1) - q1.se; transaction(p1_plus_p2 / 2); } } void subtask_3(int n, int p0) { vector<int> p(n, 0); p[0] = p0; int cnt_ones = 0; for(int i = 1; i < n - 1; i++) { typ q = transaction(p[i - 1] - 1); if(q.fi.size() == 2) { cnt_ones++; p[i] = p[i - 1] - 2; } else { p[i] = (p[i - 1] - 1) - q.se; } for(int j = 1; j < i; j++) { transaction(p[i]); } } while(cnt_ones < n - 1) { transaction(p[n - 2] - 1); cnt_ones++; } } void buy_souvenirs(signed N, int P0) { /*if(N == 2) { solve_2(N, P0); return; } if(N == 3) { solve_3(N, P0); return; }*/ subtask_3(N, P0); return; //pair<vector<signed>, int> res = transaction(3); //return; }