Kolmijako

input/code.cpp: In function 'std::__cxx11::string b3(int, int)':
input/code.cpp:10:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  while(s.size() < k) s += '0';
        ~~~~~~~~~^~~
input/code.cpp: In function 'int main()':
input/code.cpp:28:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int j = 0; j < s3.size(); j++){
                  ~~^~~~~~~~~~~

#include<bits/stdc++.h>
using namespace std;

string b3(int a, int k){
	string s = "";
	while(a > 0){
		s += '0'+a%3;
		a /= 3;
	}
	while(s.size() < k) s += '0';
	reverse(s.begin(), s.end());
	return s;
}

int main(){
	int n;
	cin >> n;

	int kolme = 1;
	for(int i = 0; i < n; i++){
		kolme *= 3;
	}
	for(int i = 0; i < kolme; i++){
		string s3 = b3(i, n);
		long long s0 = 0;
		long long s1 = 0;
		long long s2 = 0;
		for(int j = 0; j < s3.size(); j++){
			if(s3[j] == '0') s0 += j+1;
			if(s3[j] == '1') s1 += j+1;
			if(s3[j] == '2') s2 += j+1;
		}
		if(s3 == "22201"){
		cout << s0 << " " << s1 << " " << s2 << "\n";
		}
		if(s0 == s1-1 && s0 == s2-2) {
			for(auto u : s3){
				cout << (char)(u + 1) << " ";
			}
			cout << "\n";
			return 0;
		}
	}
	cout << "IMPOSSIBLE\n";
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: ACCEPTED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 1, 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 2

Verdict: TIME LIMIT EXCEEDED