Kolmijako

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

int used[1001];

void f(vector<int> &v, int &cnt, int goal, int a){
    int sum=0;
    for (int i=a; i>=1; --i){
        if (used[i]==0 && sum+i <= goal){
            cnt++;
            used[i] = true;
            v.push_back(i);
            sum = sum+i;
        }
        if (sum==goal) return;
    }
}

void g(vector<int> v, int cnt){
    cout << cnt << "\n";
    for (auto u: v){
        cout << u << " ";
    }
    cout << "\n";

}

int main(){
    
    int t;
    cin >> t;
    for (int i=1; i<=t; ++i){
        int a;
        cin >> a;
        if (a < 5 || a%3 == 1){
            cout << "NO\n";
            continue;
        }
        
        cout << "YES\n";
        int cnt1=0, cnt2=0, cnt3=0;
        vector<int> v1, v2, v3;
        for (int  i=1; i<=a; ++i) used[i] = 0;
        int goal = a*(a+1)/6;
        f(v1, cnt1, goal, a);
        f(v2, cnt2, goal, a);
        f(v3, cnt3, goal, a);
        g(v1, cnt1);
        g(v2, cnt2);
        g(v3, cnt3);
        

    }
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED