Lisäykset

#include <bits/stdc++.h>

using namespace std;

struct kohta_t {
    int lisäys;
    int määrä;
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m;
    cin >> n >> m;

    map<int, int> määrät;
    for (int i = 0; i < n; ++i) {
        int luku;
        cin >> luku;
        määrät[luku]++;
    }

    map<int, kohta_t> data;
    int kohta = 0, luku0 = 0;
    for (const auto& [luku, määrä] : määrät) {
        data[kohta] = {luku - luku0, määrä};
        kohta += määrä;
        luku0 = luku;
    }
    data[kohta] = {luku0 + m+1, 0};

    for (int i = 0; i < m; ++i) {
        int muutettavia;
        cin >> muutettavia;
        if (!muutettavia) {
            continue;
        }

        data.begin()->second.lisäys += 1;

        auto it = data.upper_bound(muutettavia);
        --it;
        it->second.lisäys -= 1;

        muutettavia -= it->first;
        if (muutettavia > 0) {
            int ei_muutettavia = it->second.määrä - muutettavia;
            data.insert({it->first + ei_muutettavia, {1, muutettavia}});
            it->second.määrä = ei_muutettavia;

            auto it2 = next(next(it));
            if (it2 != data.end()) {
                it2->second.lisäys -= 1;
                if (it2->second.lisäys == 0) {
                    prev(it2)->second.määrä += it2->second.määrä;
                    data.erase(it2);
                }
            }
        }
        if (it->second.lisäys == 0) {
            prev(it)->second.määrä += it->second.määrä;
            data.erase(it);
        }
    }

    int luku = 0;
    for (const auto& [_, kohta] : data) {
        luku += kohta.lisäys;
        for (int i = 0; i < kohta.määrä; ++i) {
            cout << luku << " ";
        }
    }
    cout << "\n";

    return 0;
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 1, 3

Verdict: ACCEPTED

Test 4

Group: 1, 3

Verdict: ACCEPTED

Test 5

Group: 1, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: ACCEPTED

Test 8

Group: 3

Verdict: ACCEPTED

Test 9

Group: 3

Verdict: ACCEPTED

Test 10

Group: 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED