Split in Three

input/code.cpp: In function 'long long int rec(long long int, long long int)':
input/code.cpp:13:13: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if (pos == v.size()) {
         ~~~~^~~~~~~~~~~
input/code.cpp: In function 'int main()':
input/code.cpp:48:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     while (p != v.size()) {
            ~~^~~~~~~~~~~
input/code.cpp:64:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     while (p != v.size()) {
            ~~^~~~~~~~~~~
input/code.cpp:80:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     while (p != v.size()) {
            ~~^~~~~~~~~~~

#include <bits/stdc++.h>

#define int long long

using namespace std;

int dp[105][5055];
int to[105][5055];
int n;
vector<int> v;
int seek;
int rec(int pos, int sum) {
    if (pos == v.size()) {
        if (sum == seek) return true;
        else return false;
    }
    if (dp[pos][sum] != -1) return dp[pos][sum];
    int res = 0;
    if (rec(pos+1,sum) == 1) {
        res = 1;
        to[pos][sum] = 0;
    } else if (rec(pos+1,sum+v[pos]) == 1) {
        res = 1;
        to[pos][sum] = 1;
    }
    dp[pos][sum] = res;
    return res;
}

signed main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cin >> n;
    int taken[n+1];
    for (int i = 0; i <= n; i++) taken[i] = 0;
    int sum = (n*(n+1))/2;
    //cout << "Sum = " << sum << '\n';
    int aSum = (sum+1)/3-1;
    int bSum = (sum+1)/3;
    int cSum = (sum+1)/3+1;
    //cout << "ASum = " << aSum << " BSum = " << bSum << " CSum = " << cSum << '\n';
    for (int i = 1; i <= n; i++) v.push_back(i);
    for (int i = 0; i < 105; i++) for (int j = 0; j < 5055; j++) dp[i][j] = -1;
    seek = aSum;
    rec(0,0);
    int p = 0;
    int s = 0;
    while (p != v.size()) {
        if (to[p][s] == 0) {
            p++;
        } else if (to[p][s] == 1) {
            s += v[p];
            taken[v[p]] = 1;
            p++;
        }
    }
    v = {};
    for (int i = 1; i <= n; i++) if (taken[i] == 0) v.push_back(i);
    for (int i = 0; i < 105; i++) for (int j = 0; j < 5055; j++) dp[i][j] = -1;
    seek = bSum;
    rec(0,0);
    p = 0;
    s = 0;
    while (p != v.size()) {
        if (to[p][s] == 0) {
            p++;
        } else if (to[p][s] == 1) {
            s += v[p];
            taken[v[p]] = 2;
            p++;
        }
    }
    v = {};
    for (int i = 1; i <= n; i++) if (taken[i] == 0) v.push_back(i);
    for (int i = 0; i < 105; i++) for (int j = 0; j < 5055; j++) dp[i][j] = -1;
    seek = cSum;
    rec(0,0);
    p = 0;
    s = 0;
    while (p != v.size()) {
        if (to[p][s] == 0) {
            p++;
        } else if (to[p][s] == 1) {
            s += v[p];
            taken[v[p]] = 3;
            p++;
        }
    }
    for (int i = 1; i <= n; i++) cout << taken[i] << ' ';
    cout << '\n';
}
/*

*/

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: WRONG ANSWER

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 1, 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: WRONG ANSWER

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 2

Verdict: WRONG ANSWER

Test 13

Group: 2

Verdict: ACCEPTED

Test 14

Group: 2

Verdict: ACCEPTED

Test 15

Group: 2

Verdict: WRONG ANSWER