Ruudukko

import sys
def pienin(lista2):
	lista2.sort()
	lista1=[]
	for i in range(1, max(lista2)+1):
		lista1.append(i)

	for i in range(len(lista2)):
		lista2.append(max(lista2)+1)
	lista2 = list(dict.fromkeys(lista2))


	for i in range(len(lista1)):
		if lista1[i] != lista2[i]:
			return lista1[i]
	return max(lista1)+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(apu2)



	for i in range(n):
		for x in range(n):
			print taulukko[i][x],
		print ""

n = input()
taulukko(n)

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