Kolmijako

input/code.cpp: In function 'bool ps(int)':
input/code.cpp:9:1: warning: no return statement in function returning non-void [-Wreturn-type]
 }
 ^

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
int arr[100];
int g[100];
int n;
bool ps(int x){

}
int main(){
    cin >> n;
    if(n == 3){
        cout << "1 2 3\n";
        exit(0);
    }

    int tsum = (((n*(n+1))/2))/3;
  /*  if(tsum%3 != 0){
        cout << "IMPOSSIBLE\n";
        exit(0);
    }
*/
  //cout << tsum << endl;


    int bins = 1;
    stringstream ss;
    int a[3]{tsum-1,tsum,tsum+1};
    //cout << tsum << endl;
    vector<int> asd;
    for(int i = n; i >= 1; i--){
        int j;

        int m = 100, bi = 0;

        for(j = 0; j < bins; j++){
            if(a[j] >= i && a[j] - i < m){
                m = a[j] - i;
                bi = j;
            }
        }

        if(m == 100){
            if(bins == 3){
                cout << "IMPOSSIBLE\n" << endl;
                exit(0);
                break;
            } else {
                asd.push_back(bins+1);
                a[bins]-=i;
                bins++;
            }
        } else {
            a[bi] -= i;
            asd.push_back(bi+1);
        }
    }
    if(a[0] == a[1] && a[0] == a[2] && a[0] == 0){
        for(int i = n-1; i >= 0; i--){
            cout << asd[i] << " ";
        }
        cout << endl;
    } else {
        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: WRONG ANSWER

Test 10

Group: 2

Verdict: WRONG ANSWER

Test 11

Group: 2

Verdict: WRONG ANSWER

Test 12

Group: 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: WRONG ANSWER

Test 14

Group: 2

Verdict: WRONG ANSWER

Test 15

Group: 2

Verdict: ACCEPTED