#include "festival.h" #include <bits/stdc++.h> #define all(x) (x).begin(), (x).end() using namespace std; std::vector<int> max_coupons(int A, std::vector<int> P, std::vector<int> T) { vector<int> Z; vector<pair<int, int>> C; for (int i = 0; i < P.size(); i++) { C.push_back({i, P[i]}); } sort(all(C), [](const auto &a, const auto &b) { return a.second > b.second; }); int i = 0; while (A > 0 && i < (int)C.size()){ if (A - C[i].second >= 0){ A -= C[i].second; Z.push_back(C[i].first); } i++; } return Z; }