Submission #1251491

#	Time	Username	Problem	Language	Result	Execution time	Memory
1251491		vietbachleonkroos2326	Festival (IOI25_festival)	C++20	5 / 100	41 ms	5564 KiB

festival

#include <bits/stdc++.h>
using namespace std;

int n;
vector<int> res;

void solvesub1(int &A, vector<int> &P, vector<int> &T) {
    vector<pair<int, int>> vec;
    for (int i = 0; i < n; i++) {
        vec.push_back({P[i], i});
    }
    
    sort(vec.begin(), vec.end());
    
    for (int i = 0; i < n; i++) {
        if (A >= vec[i].first) {
            A -= vec[i].first;
            res.push_back(vec[i].second);
        } else {
            break;
        }
    }
}

void solvesub2(int &A, vector<int> &P, vector<int> &T) {
    vector<pair<long long, int>> vec; 
    
    for (int i = 0; i < n; i++) {
        if (T[i] == 2) {
            vec.emplace_back(P[i], i);
        } else {
            vec.emplace_back(1e9 + P[i], i);
        }
    }
    
    sort(vec.begin(), vec.end());
    
    for (auto [_, i] : vec) {
        if (A >= P[i]) {
            res.push_back(i);
            A = (A - P[i]) * T[i];
        }
    }
}

void solvesub3(int &A, vector<int> &P, vector<int> &T) {
    
}

vector<int> max_coupons(int A, vector<int> P, vector<int> T) { 
    n = T.size();
    res.clear();
    
    bool checksub1 = true;
    bool checksub2 = true;
    for (int i = 0; i < n; i++) {
        if (T[i] != 1) checksub1 = false;
        if (T[i] > 2) checksub2 = false;
    }

    if (checksub1) solvesub1(A, P, T);
    else if (checksub2) solvesub2(A, P, T);
    else solvesub3(A, P, T);

    return res;
}

• Subtask 0: 0 / 0
• Subtask 1: 5 / 5
• Subtask 2: 0 / 7
• Subtask 3: 0 / 12
• Subtask 4: 0 / 15
• Subtask 5: 0 / 27
• Subtask 6: 0 / 16
• Subtask 7: 0 / 18