Lisäykset

#pragma GCC optimize("O3,unroll-loops")
#include <algorithm>
#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];
	}

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

	for (; ;) {
		bool f = 0;
		for (int i = 0; i < n - 1; ++i) {
			if (v[i] > v[i + 1]) {
				int j = i + 1;
				while (j + 1 < n && v[j] + 1 > v[j + 1]) ++j;
				long s = 0;
				for (int k = i; k <= j; ++k) {
					s += v[k];
				}
				for (int k = i; k <= j; ++k) {
					v[k] = s / (j - i + 1) + (j - k < s % (j - i + 1));
				}
				f = 1;
				i = j;
			}
		}
		if (!f) break;
	}

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

Test 6

Group: 2, 3

Verdict: ACCEPTED

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

Test 11

Group: 3

Verdict: ACCEPTED