#include <bits/stdc++.h>
#include "festival.h"
using namespace std;
using ll = long long;
std::vector<int> max_coupons(int A, std::vector<int> P, std::vector<int> T) {
    vector<pair<int, int>> v1, v2;
    int n = (int)P.size();
    for (int i = 0; i < n; i++) {
        if (T[i] == 1) {
            v1.push_back({P[i], i});
        } else {
            v2.push_back({P[i], i});
        }
    }
    vector<int> ans;
    sort(v1.begin(), v1.end());
    sort(v2.begin(), v2.end());
    for (auto [x, i] : v2) {
        if ((A - x) * 2 >= A) {
            A = (A - x) * 2;
            ans.push_back(i);
        }
    }
    int l1 = 0, l2 = 0;
    while (true) {
        int a = 1e9 + 1, b = 1e9 + 1;
        if (v1.size() > l1) {
            a = v1[l1].first;
        }
        if (v2.size() > l2) {
            b = A - (A - v2[l2].first) * 2;
        }
        if (min(a, b) > A) {
            break;
        }
        if (a < b) {
            ans.push_back(v1[l1].second);
            l1++;
        } else {
            ans.push_back(v1[l2].second);
            l2++;
        }
    }
    return ans;
}
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... |