Ruudukko

import sys
def pienin(lista1,lista2):
	lista2.sort()
	lista2 = list(dict.fromkeys(lista2))
	for i in range(len(lista2),len(lista1)):
		lista2.append(0)
 
	for i in range(len(lista1)):
		if lista1[i] != lista2[i]:
			return lista1[i]
	return max(lista2)+1
 
 
 
def taulukko(n):
	taulukko = []
	for i in range(n):
		taulukko.append([]) #fill array with arrays
	
	for i in range(n):
		for x in range(n):
			taulukko[i].append(None)
 
	
 
	for rivi in range(n):
		apu = [i for i in range(1,n+1)]
		if rivi == 0:
			taulukko[rivi]= apu
		else:
 
			for sarake in range(n):
				apu = [i for i in range(1,n+1)]
				apu2=[]
				
				for i in range(rivi):
					apu2.append(taulukko[i][sarake])
				for i in range(sarake):
					apu2.append(taulukko[rivi][i])
				taulukko[rivi][sarake]=pienin(apu,apu2)
 
 
 
	for i in range(n):
		for x in range(n):
			if x < n-1:
				sys.stdout.write(str(taulukko[i][x])+ " ")
			else:
				sys.stdout.write(str(taulukko[i][x])+ "\n")
 
n = input()
taulukko(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