#include <bits/stdc++.h> using namespace std; pair<vector<int>, long long> transaction(long long M); // ну сколько можно ебаный рот блять че тебе еще надо void buy_souvenirs(int N, long long P0) { vector<int> bought(N, 0); int total_needed = (N - 1) * N / 2; int total_bought = 0; long long M = P0 - 1; while (total_bought < total_needed) { auto res = transaction(M); auto &L = res.first; for (int type : L) { if (type == 0) continue; if (bought[type] < type) { bought[type]++; total_bought++; } } } }