Split in Three

input/code.cpp: In function 'void usaco(std::__cxx11::string)':
input/code.cpp:12:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   freopen((s+".in").c_str(), "r", stdin);
   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
input/code.cpp:13:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   freopen((s+".out").c_str(), "w", stdout);
   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#include <bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for(int i = a; i < (b); ++i)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
typedef long long ll;
typedef pair<ll, ll> pii;
typedef vector<ll> vi;
typedef vector<pii> vpi;

void usaco(string s){
  freopen((s+".in").c_str(), "r", stdin);
  freopen((s+".out").c_str(), "w", stdout);
}

int main() {
	cin.tie(0)->sync_with_stdio(0);
	cin.exceptions(cin.failbit);
#ifdef LOCAL_DEFINE
	freopen("input.txt", "r", stdin);
#endif
  int n; cin >> n;
  vector<int> v, ans(n, 0);
  for(int i = 1; i <= n; i++) v.push_back(i);
  int tot =  (n*(n+1))/2;
  int sum1 = tot/3 - 1, sum2 = tot/3, sum3 = tot/3 + 1, done = 0;
  //cout << tot << ' ' << sum1 << ' ' << sum2 << ' ' << sum3 << '\n';
  if(sum1+sum2+sum3!=tot or sum1 == 0){
    cout << "IMPOSSIBLE\n";
    return 0;
  }

  do{
  	int s1=0,s2=0,s3=0;// done=0;
  	done = 0;
  	for(int i = 0; i < n; i++){
  	  if(s1!=sum1){
  	    s1 += v[i];
  	    ans[v[i]-1] = 1;
  	  }
  	  else if(s2!=sum2){
  	    s2 += v[i];
  	    ans[v[i]-1] = 2;
  	  }
  	  else if(s3!=sum3){
  	    s3 += v[i];
  	    ans[v[i]-1] = 3;
  	  }
  	  
  	  if(s3 == sum3){
  	    done = 1;
  	  }
  	}
  	if(done){
  	  for(auto i: ans) cout << i << ' ';
  	  cout << '\n';
  	  break;
  	}
  }
  while(next_permutation(all(v)));
  if(!done or sum1+sum2+sum3!=tot or sum1 == 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: 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