154372025-02-19 16:11:53feheristvanÓvodacpp17Wrong answer 5/50101ms3524 KiB
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

struct Gyerek {
    int akar_szerep, siras, index;
};

struct Szerep {
    int db, index, mdb;
};

bool comp(const Gyerek &a, const Gyerek &b) {
    return a.siras < b.siras;
}

int main() {
    int n, m;
    cin >> n >> m;

    vector<Gyerek> ovo(n);
    vector<Szerep> sz(m);

    for (int i = 0; i < m; i++) {
        cin >> sz[i].db;
        sz[i].mdb = sz[i].db;
        sz[i].index = i;
    }

    for (int i = 0; i < n; i++) {
        cin >> ovo[i].akar_szerep;
        ovo[i].akar_szerep--; // Adjusting for 0-based indexing
        ovo[i].index = i;
    }

    for (int i = 0; i < n; i++) {
        cin >> ovo[i].siras;
    }

    sort(ovo.begin(), ovo.end(), comp); // Sort children by crying level

    vector<int> kapott_szerep(n, -1);
    int sirok = 0;

    for (int i = 0; i < n; i++) {
        int preferred = ovo[i].akar_szerep;

        if (sz[preferred].db > 0) {
            sz[preferred].db--;
            kapott_szerep[ovo[i].index] = sz[preferred].index;
        } else {
            sirok += ovo[i].siras;
            bool assigned = false;

            for (int j = 0; j < m; j++) {
                if (sz[j].db > 0) {
                    sz[j].db--;
                    kapott_szerep[ovo[i].index] = sz[j].index;
                    assigned = true;
                    break;
                }
            }

            if (!assigned) {
                kapott_szerep[ovo[i].index] = -1; // No role available
            }
        }
    }

    cout << sirok << endl;
    for (int i : kapott_szerep) {
        cout << i + 1 << " "; // Convert back to 1-based indexing
    }
    cout << endl;

    return 0;
}

SubtaskSumTestVerdictTimeMemory
base5/50
1Wrong answer0/01ms316 KiB
2Wrong answer0/04ms500 KiB
3Wrong answer0/21ms316 KiB
4Partially correct1/21ms316 KiB
5Wrong answer0/21ms316 KiB
6Accepted2/21ms316 KiB
7Wrong answer0/21ms316 KiB
8Wrong answer0/21ms316 KiB
9Accepted2/21ms316 KiB
10Wrong answer0/21ms316 KiB
11Wrong answer0/21ms316 KiB
12Wrong answer0/21ms316 KiB
13Wrong answer0/22ms500 KiB
14Wrong answer0/32ms316 KiB
15Wrong answer0/312ms640 KiB
16Wrong answer0/363ms1096 KiB
17Wrong answer0/337ms1212 KiB
18Wrong answer0/361ms2192 KiB
19Wrong answer0/368ms2100 KiB
20Wrong answer0/379ms2100 KiB
21Wrong answer0/397ms2528 KiB
22Wrong answer0/4101ms3524 KiB