Ruudukko

#include <iostream>
#include <fstream>
#include <algorithm>

using namespace std;

int main() {
    int n;
    cin >> n;
    int a[n][n];
    int k = 1;
    for(int j = 0; j < n; ++j) {
        int I = 0, J = j;
        while(I < n && J >= 0) {
            a[I][J] = k;
            ++I, --J;
        }
        k++;
    }
    k = 1;
    for(int i = 1; i < n; ++i) {
        int I = i, J = n - 1;
        while(I < n && J >= 0) {
            a[I][J] = k;
            ++I, --J;
        }
        k++;
    }
    for(int i = 0; i < n; ++i) {
        for(int j = 0; j < n; ++j) {
            cout << a[i][j] << " ";
        }
        cout << endl;
    }
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER