Ruudukko

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

int n;
vector<vector<int>> ve;
int main(){
  cin >> n;
  if(n == 1){
    cout << 1 << "\n";
  }else if(n == 2){
    cout << 1 << " " << 2 << "\n";
    cout << 2 << " " << 1 << "\n";
  }else if(n == 3){
    cout << 1 << " " << 2 << " " << 3 << "\n";
    cout << 2 << " " << 1 << " " << 4 << "\n";
    cout << 3 << " " << 4 << " " << 1 << "\n";
  }
  if(n >= 4){
    ve.push_back({1,2,3,4});
    ve.push_back({2,1,4,3});
    ve.push_back({3,4,1,2});
    ve.push_back({4,3,2,1});
    int a = 4;
    while(a < n){
      for(int i = 0; i < a; i++){
          ve[i].push_back(a+i+1);
      }
      ve.push_back({});
      for(int j = 0; j <= a; j++){
        if(j != a){
        ve[ve.size()-1].push_back(a+j+1);
      }else{
        ve[ve.size()-1].push_back(1);
      }
      }
      a++;
    }
  }
  for(auto u : ve){
    for(auto x : u){
      cout << x << " ";
    }
    cout << "\n";
  }
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER