Lisäykset

#include <climits>
#include <cstdint>
#include <iostream>
#include <map>
#include <vector>

using namespace std;

int n, m;
map<int, int> erotukset;

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

    cin >> n >> m;

    int edellinen = -1010101010;
    for(int i = 0; i <= n; ++i) {
        int x;
        if(i == n) {
            x = 1010101010;
        } else {
            cin >> x;
        }
        if(x != edellinen) {
            erotukset[i] = x - edellinen;
        }
        edellinen = x;
    }

    auto kasvata = [&](int i) {
        ++erotukset[i];
    };
    auto pienenna = [&](int i) {
        auto it = erotukset.find(i);
        if(it == erotukset.end()) {
            throw 5;
        }
        if(it->second == 1) {
            erotukset.erase(it);
        } else {
            if(it->second < 1) {
                throw 5;
            }
            --it->second;
        }
    };

    for(int i = 0; i < m; ++i) {
        int k;
        cin >> k;

        auto it = erotukset.lower_bound(k);
        if(it == erotukset.end()) throw 5;
        if(it->first == k) {
            pienenna(k);
        } else {
            if(it == erotukset.begin()) throw 5;
            auto it2 = it;
            --it2;

            int a = it2->first;
            int b = it->first;

            pienenna(b);
            kasvata(b - (k - a));
            pienenna(a);
        }
    }

    vector<int> tulos(n);
    int luku = 1010101010 - erotukset[n];
    for(int i = n - 1; i >= 0; --i) {
        tulos[i] = luku;
        luku -= erotukset[i];
    }
    for(int x : tulos) {
        cout << x << " ";
    }
    cout << "\n";

    return 0;
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

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: ACCEPTED

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