Summat

input/code.cpp: In function 'bool ratkaise(int, int, long int, int, std::vector<int>&, std::unordered_map<int, int>)':
input/code.cpp:15:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for (int i = 1; i < luvut.size(); ++i) {
                    ~~^~~~~~~~~~~~~~
input/code.cpp:25:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    if (luvut_yhteensa < (n - luvut.size()) * luvut_keskiarvo) {
        ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
 
bool ratkaise(int a, int n, long luvut_yhteensa, int luvut_keskiarvo, std::vector<int>& summat, std::unordered_map<int, int> summat_n) {
	std::vector<int> luvut {a};
	luvut_yhteensa -= a;
	std::unordered_map<int, int> tarjolla;
	for (int summa: summat) {
		if (tarjolla[summa]) {
			tarjolla[summa] -= 1;
		} else {
			int b = summa - luvut[0];
			for (int i = 1; i < luvut.size(); ++i) {
				int c = b + luvut[i];
				if (!summat_n[c]) {
					return false;
				}
				summat_n[c] -= 1;
				tarjolla[c] += 1;
			}
			luvut.push_back(b);
			luvut_yhteensa -= b;
			if (luvut_yhteensa < (n - luvut.size()) * luvut_keskiarvo) {
				return false;
			}
		}
	}
	for (int luku: luvut) {
		std::cout << luku << ' ';
	}
	std::cout << '\n';
	return true;
}
 
int main() {
	int n;
	std::cin >> n;
	std::vector<int> summat(n * (n-1) / 2);
	std::unordered_map<int, int> summat_n;
	long summat_yhteensa = 0;
	for (auto& k: summat) {
		std::cin >> k;
		summat_n[k] += 1;
		summat_yhteensa += k;
	}
	std::sort(summat.begin(), summat.end());
	long luvut_yhteensa = summat_yhteensa / (n-1);
	int luvut_keskiarvo = luvut_yhteensa / n;
 
	for (int a = 1; a <= luvut_keskiarvo; ++a) {
		if (ratkaise(a, n, luvut_yhteensa, luvut_keskiarvo, summat, summat_n)) {
			break;
		}
	}
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

Test 9

Group: 2, 3

Verdict: ACCEPTED

Test 10

Group: 2, 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED