Как сгенерировать список всех различных латинских квадратов 3x3 в Python
Латинский квадрат - это массив размером nxn, заполненный n разными символами, каждый из которых встречается ровно один раз в каждой строке и ровно один раз в каждом столбце (например, судоку). Пример латинского квадрата:
1 2 3
2 3 1
3 1 2
Это то, что я пробовал, но все еще не все отчетливо
grid = []
temp = []
block = [[1,2,3],
[2,3,1],
[3,1,2]]
perm = permutations(block)
for i in perm: #row permutations
temp.extend(i)
if len(temp)==3:
grid.extend([temp])
temp = []
for perm in zip(permutations(block[0]), permutations(block[1]), permutations(block[2])): #column permutations
temp.extend([perm])
for i in range(len(temp)): #convert to list
temp[i] = list(temp[i])
for j in range(len(temp[0])):
temp[i][j] = list(temp[i][j])
grid.extend(temp)
for i in grid:
for j in i:
print(j)
print()
Выход:
[1, 2, 3]
[2, 3, 1]
[3, 1, 2]
[1, 2, 3]
[3, 1, 2]
[2, 3, 1]
[2, 3, 1]
[1, 2, 3]
[3, 1, 2]
[2, 3, 1]
[3, 1, 2]
[1, 2, 3]
[3, 1, 2]
[1, 2, 3]
[2, 3, 1]
[3, 1, 2]
[2, 3, 1]
[1, 2, 3]
[3, 1, 2]
[2, 3, 1]
[1, 2, 3]
[3, 2, 1]
[2, 1, 3]
[1, 3, 2]
[1, 3, 2]
[3, 2, 1]
[2, 1, 3]
[1, 2, 3]
[3, 1, 2]
[2, 3, 1]
[2, 3, 1]
[1, 2, 3]
[3, 1, 2]
[2, 1, 3]
[1, 3, 2]
[3, 2, 1]
Результат должен быть таким (порядок не имеет значения): Латинский квадрат
[1, 2, 3]
[2, 3, 1]
[3, 1, 2]
[1, 2, 3]
[3, 1, 2]
[2, 3, 1]
[1, 3, 2]
[2, 1, 3]
[3, 2, 1]
[1, 3, 2]
[3, 2, 1]
[2, 1, 3]
[2, 1, 3]
[1, 3, 2]
[3, 2, 1]
[2, 1, 3]
[3, 2, 1]
[1, 3, 2]
[2, 3, 1]
[1, 2, 3]
[3, 1, 2]
[2, 3, 1]
[3, 1, 2]
[1, 2, 3]
[3, 2, 1]
[1, 3, 2]
[2, 1, 3]
[3, 2, 1]
[2, 1, 3]
[1, 3, 2]
[3, 1, 2]
[1, 2, 3]
[2, 3, 1]
[3, 1, 2]
[2, 3, 1]
[1, 2, 3]
1 ответ
Решение
Вы можете использовать рекурсию с генератором:
def row(n, r, c = []):
if len(c) == n:
yield c
for i in range(1, n+1):
if i not in c and i not in r[len(c)]:
yield from row(n, r, c+[i])
def to_latin(n, c = []):
if len(c) == n:
yield c
else:
for i in row(n, [[]]*n if not c else list(zip(*c))):
yield from to_latin(n, c+[i])
for i in to_latin(3):
for b in i:
print(b)
print('-'*9)
Выход:
[1, 2, 3]
[2, 3, 1]
[3, 1, 2]
---------
[1, 2, 3]
[3, 1, 2]
[2, 3, 1]
---------
[1, 3, 2]
[2, 1, 3]
[3, 2, 1]
---------
[1, 3, 2]
[3, 2, 1]
[2, 1, 3]
---------
[2, 1, 3]
[1, 3, 2]
[3, 2, 1]
---------
[2, 1, 3]
[3, 2, 1]
[1, 3, 2]
---------
[2, 3, 1]
[1, 2, 3]
[3, 1, 2]
---------
[2, 3, 1]
[3, 1, 2]
[1, 2, 3]
---------
[3, 1, 2]
[1, 2, 3]
[2, 3, 1]
---------
[3, 1, 2]
[2, 3, 1]
[1, 2, 3]
---------
[3, 2, 1]
[1, 3, 2]
[2, 1, 3]
---------
[3, 2, 1]
[2, 1, 3]
[1, 3, 2]
---------