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) {
			int u = v[i] - v[i + 1];
			if (u > 0) {
				u = (u + 1) / 2;
				v[i] -= u;
				v[i + 1] += u;
				f = 1;
			}
		}
		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: 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: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED