Kolmijako

#include <bits/stdc++.h>

typedef long long ll;

using namespace std;

vector<vector<int>> solve2(int n, int target) {
  vector<int> sum(3);
  vector<vector<int>> mem(3);
  vector<bool> taken(n + 1);
  for (int j = 0; j < 3; j++) {
    for (int i = n; i >= 1; i--) {
      if (taken[i])
        continue;
      if (sum[j] + i <= target) {
        mem[j].push_back(i);
        sum[j] += i;
        taken[i] = true;
      }
    }
  }
  return mem;
}

void solve(int n) {
  ll sum = n * (n + 1) / 2;
  if (sum % 3 != 0 || n < 4) {
    cout << "NO\n";
  } else {
    // cout << n << " " << (sum%3==0) << " sum: " << sum  << " /3=" << sum/3<< "
    // " <<solve2(n, sum/3) << endl;
    auto ans = solve2(n, sum / 3);
    cout << "YES\n";
    for (auto u : ans) {
      cout << u.size() << "\n";
      for (auto u2 : u)
        cout << u2 << " ";
      cout << "\n";
    }
  }
}

int main() {
	int t;
	cin >> t;
	while(t--) {
		int n;
		cin >> n;
		solve(n);
	}
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED