Submission #867485

#	Time	Username	Problem	Language	Result	Execution time	Memory
867485		pizzamoeger	Zalmoxis (BOI18_zalmoxis)	C++14	100 / 100	117 ms	25900 KiB

zalmoxis

#include <bits/stdc++.h>
#define int long long

using namespace std;

signed main () {
	cin.tie(0);
	ios_base::sync_with_stdio(0);

	int n, k; cin >> n >> k;
	vector<int> S (n);
	for (int i = 0; i < n; i++) cin >> S[i];
	n++;
	S.push_back(30);

	stack<int> nums;
	vector<pair<int,int>> inserts;
	int i = 0;
	while (i < n) {
		int cur = S[i];
		while (!nums.empty()) {
			int top = nums.top();
			//cerr << cur << " " << top << " " << i << "\n";
			if (cur == top) {
				cur++;
				nums.pop();
			} else if (cur > top) {
				inserts.push_back({i, top});
				cur = top+1;
				nums.pop();
				i--;
			} else {
				nums.push(cur);
				break;
			}
		}
		if (nums.empty()) 	nums.push(cur);
		i++;
	}

	if (!nums.empty() && nums.top() != 31) inserts.push_back({n, nums.top()});

	int left = k-inserts.size();
	int j = 0;
	while (left > 0) {
		while (inserts[j].second == 0) j++;
		inserts[j].second--;
		inserts.push_back(inserts[j]);
		left--;
	}

	sort(inserts.begin(), inserts.end());

	j = 0;
	for (int i = 0; i < n; i++) {
		while (j < inserts.size() && i == inserts[j].first) {
			int cnt = inserts[j].second;
			cout << cnt << " ";
			j++;
		}
		if (i != n-1) cout << S[i] << " ";
	}
	while (j < inserts.size() && n == inserts[j].first) {
		assert(false);
		int cnt = inserts[j].second;
		cout << cnt << " ";
		j++;
	}
	cout << "\n";
}

Compilation message (stderr)

zalmoxis.cpp: In function 'int main()':
zalmoxis.cpp:56:12: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   56 |   while (j < inserts.size() && i == inserts[j].first) {
      |          ~~^~~~~~~~~~~~~~~~
zalmoxis.cpp:63:11: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |  while (j < inserts.size() && n == inserts[j].first) {
      |         ~~^~~~~~~~~~~~~~~~

• Subtask 1: 30 / 30
• Subtask 2: 70 / 70