#include <bits/stdc++.h>
#include "festival.h"
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;
#define all(v) v.begin(), v.end()
constexpr ll inf = 1ll << 62ll;
mt19937 mt(34056237);
ll _ = 0;
vector<int> max_coupons(int A, vector<int> price, vector<int> type) {
ll n = price.size();
ll mx_type = *max_element(all(type));
vector<vector<pll>> coups(mx_type);
for (ll i = 0; i < n; i++) {
coups[type[i]].emplace_back(price[i], i);
}
for (ll i = 0; i < mx_type; i++) sort(all(coups[i]));
vector<int> res;
for (ll i = 0; i < coups[0].size(); i++) {
A -= coups[0][i].first;
if (A < 0) break;
res.push_back(coups[0][i].second);
}
return res;
}
