Lisäykset

#include <iostream>
#include <vector>

using namespace std;

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

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

	vector<int> v(n);
	for (int i = 0; i < n; ++i) {
		cin >> v[i];
	}

	vector<int> t(n);
	for (int i = 0; i < m; ++i) {
		int x;
		cin >> x;
		t[x - 1]++;
	}

	for (int i = n - 1; i > 0; --i) {
		t[i - 1] += t[i];
	}

	auto f = [&](int l, int r, int x) {
		long b = 0;
		for (int i = l; i < r; ++i) {
			b += x - v[i] - t[i];
		}
		return b <= 0;
	};

	for (int i = 0, j = 0; i < n; i = j) {
		j = i + 1;
		while (j < n && f(i, j, v[j])) {
			for (int k = i; k < j; ++k) {
				t[k] -= v[j] - v[k];
				v[k] = v[j];
			}
			++j;
		}

		long s = 0;
		for (int k = i; k < j; ++k) {
			s += t[k];
			t[k] = 0;
		}
		for (int k = i; k < j; ++k) {
			v[k] += s / (j - i);
		}
		s %= j - i;
		for (int k = j - 1; s && k >= i; --k) {
			v[k] += 1;
			s--;
		}
	}

	for (int i = 0; i < n; ++i) {
		cout << v[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: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED