Ruudukko

n=int(input())
 
array = []
array.append(list(range(1,n+1)))
#print(array)
 
for i in range(n-1):
#    print("i",i)
    row=[]
    for j in range(n):
       # print("j",j)
        c=1
        while True:
      #      print("c",c)
            valid = False
            if c not in row:
                valid = True
                for ii in range(i+1):
     #               print("ii",ii)
                    if c == array[ii][j]:
                        valid = False
    #                    print("column")
              #      else:
   #                     print("iij",ii,j)
#            else:
  #              print("row")
            if valid:
 #               print("chose",c)
                row.append(c)
                break
            c+=1
#    print(row)
    array.append(row)
for i in array:
    print(" ".join(map(str,i)))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: TIME LIMIT EXCEEDED

Test 6

Verdict: TIME LIMIT EXCEEDED