Lisäykset

#include <cassert>
#include <iostream>
#include <map>
#include <unordered_map>
#include <string>
#include <sstream>

void print(std::map<int, int>& luvut)
{
	std::string str;
	for (const auto& [luku, maara] : luvut)
	{
		for (int i = 0; i < maara; ++i)
			str += std::to_string(luku) + " ";
	}
	std::cout << str << std::endl;

}

void increase(std::map<int, int>& luvut, int k)
{
	auto it = luvut.begin();
	int luku = it->first;
	int maara = it->second;

	int sub = k > maara ? maara : k;
	k -= sub;
	it->second -= sub;

	if (it->second == 0)
		luvut.erase(it);

	if (k > 0)
		increase(luvut, k);
	luvut[luku + 1] += sub;

}

int main(void)
{
	int n, m;
	std::cin >> n >> m;
	std::cin.ignore();

	std::string alkusis;
	std::getline(std::cin, alkusis);
	std::stringstream ss(alkusis);

	std::map<int, int> luvut;
	int luku;
	while (ss >> luku)
		luvut[luku]++;

	std::string ope;
	std::getline(std::cin, ope);
	std::stringstream sope(ope);

	int k;
	while (sope >> k)
		increase(luvut, k);

	print(luvut);
}

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