Lisäykset

#include <bits/stdc++.h>

using namespace std;

int cnt[1111111];

int main () {
    int n, m;
    cin>>n>>m;
    for (int i = 0; i < n; i++) {
        int x;
        cin>>x;
        cnt[x]++;
    }
    set<int> ne;
    for (int i = 0; i <= 1000000; i++) {
        if (cnt[i]) ne.insert(i);
    }
    for (int i = 0; i < m; i++) {
        int prev = 0;
        int previ = 0;
        int k;
        vector<int> ops;
        cin>>k;
        //cout<<"NEW "<<k<<endl;
        for (int i : ne) {
            //cout<<"i "<<i<<endl;
            if (previ + 1 != i) prev = 0;
            if (cnt[i] - prev >= k) {
                //cout<<"MOVE "<<i<<" "<<(k)<<endl;
                cnt[i] -= k;
                cnt[i + 1] += k;
                ops.push_back(i);
                ops.push_back(i + 1);
                break;
            } else {
                //cout<<"MOVE "<<i<<" "<<(cnt[i] - prev)<<endl;
                cnt[i + 1] += cnt[i] - prev;
                k -= cnt[i] - prev;
                int x = cnt[i];
                cnt[i] = prev;
                prev = x - prev;
                ops.push_back(i);
                ops.push_back(i + 1);
            }
            previ = i;
        }
        for (int i : ops) {
            if (cnt[i]) ne.insert(i);
            else ne.erase(i);
        }
    }
    for (int i = 0; i < 1111111; i++) {
        while (cnt[i]--) cout<<i<<" ";
    }
    cout<<endl;
}

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: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED