Split in Three

input/code.cpp: In function 'void check(std::vector<long long int>&)':
input/code.cpp:18:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (; id < v.size(); id++) {
            ~~~^~~~~~~~~~
input/code.cpp:28:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (; id < v.size(); id++) {
            ~~~^~~~~~~~~~
input/code.cpp:39:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (; id < v.size(); id++) {
            ~~~^~~~~~~~~~

#include <bits/stdc++.h>
 
using namespace std;
 
#define int long long
#define endl '\n'
#define maxn 200010
const int MOD = 1000000007;
 
int need;
void check(vector<int> &v) {
    int id = 0, sum = 0;
    bool fd1 = false;
    bool fd2 = false;
    bool fd3 = false;
 
    vector<int> now(v.size(), 1);
    for (; id < v.size(); id++) {
        sum += v[id];
        if (sum == need - 1) {
            fd1 = 1;
            break;
        }
    }
    id++;
 
    sum = 0;
    for (; id < v.size(); id++) {
        sum += v[id];
        now[v[id] - 1] = 2;
        if (sum == need) {
            fd2 = 1;
            break;
        }
    }
    id++;
 
    sum = 0;
    for (; id < v.size(); id++) {
        sum += v[id];
        now[v[id] - 1] = 3;
        if (sum == need + 1) {
            fd3 = 1;
            break;
        }
    }
    if (fd1 and fd2 and fd3) {
        for (auto x: now)cout << x << " ";
        exit(0);
    }
}
 
void solve() {
    int n;
    cin >> n;
 
    vector<int> v(n);
    iota(v.begin(), v.end(), 1ll);
 
//    v = {1, 3, 7, 4, 8, 2, 5, 6};
    if (n * (n + 1) % 6) {
        cout << "IMPOSSIBLE" << endl;
        return;
    }
    need = n * (n + 1) / 6;
    do {
        check(v);
    } while (next_permutation(v.begin(), v.end()));
    cout << "IMPOSSIBLE" << endl;
}
 
int32_t main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
 
    int t = 1;
//    cin >> t;
 
    while (t--)
        solve();
    return 0;
}

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: ACCEPTED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 2

Verdict: ACCEPTED