Kolmijako

#include<iostream>
#include<vector>
#include<set>
using namespace std;

void solve() {
	int n;
	cin >> n;

	int target = (n*(n+1))/6;

	vector<int> a,b,c;

	vector<int> sums(3, 0);
	for (int i = n; i > 0; i--)  {
		if (sums[2] + i <= target) {
			c.push_back(i);
			sums[2] += i;
			continue;
		}
		if (sums[1] + i <= target) {
			b.push_back(i);
			sums[1] += i;
			continue;
		}
		if (sums[0] + i <= target) {
			a.push_back(i);
			sums[0] += i;
			continue;
		}
		cout << "NO\n";
		return;
	}

	cout << "YES\n";
	cout << a.size() << "\n";
	for (auto u : a) cout << u << " ";
	cout << "\n";
	cout << b.size() << "\n";
	for (auto u : b) cout << u << " ";
	cout << "\n";
	cout << c.size() << "\n";
	for (auto u : c) {
		cout << u << " ";
	}
	cout << "\n";
}

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

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED