Ruudukko

i = int(input())

def printstuff(arr):

    output = ""
    for row in arr:
        output += " ".join(str(x) for x in row)
        output += "\n"

    print(output.strip());


def stuff():
    rows = [];
    initialRow = [];
    for x in range(1, i + 1):
        initialRow.append(x)

    rows.append(initialRow.copy())
    for row in range(0, i - 1):
        initialRow.insert(0, initialRow.pop(i - 1))
        rows.append(initialRow.copy())

    for row in rows:
        row.append("\n")

    printstuff(rows)
stuff()

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