N-queen backtracking em Python: como devolver as soluções em vez de imprimi-las?
def solve(n):
#prepare a board
board = [[0 for x in range(n)] for x in range(n)]
#set initial positions
place_queen(board, 0, 0)
def place_queen(board, row, column):
"""place a queen that satisfies all the conditions"""
#base case
if row > len(board)-1:
print board
#check every column of the current row if its safe to place a queen
while column < len(board):
if is_safe(board, row, column):
#place a queen
board[row][column] = 1
#place the next queen with an updated board
return place_queen(board, row+1, 0)
else:
column += 1
#there is no column that satisfies the conditions. Backtrack
for c in range(len(board)):
if board[row-1][c] == 1:
#remove this queen
board[row-1][c] = 0
#go back to the previous row and start from the last unchecked column
return place_queen(board, row-1, c+1)
def is_safe(board, row, column):
""" if no other queens threaten a queen at (row, queen) return True """
queens = []
for r in range(len(board)):
for c in range(len(board)):
if board[r][c] == 1:
queen = (r,c)
queens.append(queen)
for queen in queens:
qr, qc = queen
#check if the pos is in the same column or row
if row == qr or column == qc:
return False
#check diagonals
if (row + column) == (qr+qc) or (column-row) == (qc-qr):
return False
return True
solve(4)
Eu escrevi código Python para o problema N-queen e imprime todas as soluções possíveis sempre que é encontrado. No entanto, gostaria de modificar esse código para que ele retorne uma lista de todas as soluções (configurações de placa) em vez de imprimi-las. Eu tentei modificar o código da seguinte forma:
def solve(n):
#prepare a board
board = [[0 for x in range(n)] for x in range(n)]
#set initial positions
solutions = []
place_queen(board, 0, 0, solutions)
def place_queen(board, row, column, solutions):
"""place a queen that satisfies all the conditions"""
#base case
if row > len(board)-1:
solutions.append(board)
return solutions
#check every column of the current row if its safe to place a queen
while column < len(board):
if is_safe(board, row, column):
#place a queen
board[row][column] = 1
#place the next queen with an updated board
return place_queen(board, row+1, 0, solutions)
else:
column += 1
#there is no column that satisfies the conditions. Backtrack
for c in range(len(board)):
if board[row-1][c] == 1:
#remove this queen
board[row-1][c] = 0
#go back to the previous row and start from the last unchecked column
return place_queen(board, row-1, c+1, solutions)
No entanto, isso retorna assim que a primeira solução é encontrada, entãosolutions
consiste apenas em uma possível configuração da placa. Desde aimpressão vs. retorno Foi confuso para mim para backtracking algoritmos, eu agradeceria muito se alguém poderia me mostrar como modificar o código acima, e como abordar problemas semelhantes no futuro.
Eu assumo que usar uma variável global funcionaria, mas aprendi de algum lugar que usar variáveis globais para tal problema é desencorajado. Alguém poderia explicar isso também?
EDITAR:
def solve(n):
#prepare a board
board = [[0 for x in range(n)] for x in range(n)]
#set initial positions
solutions = list()
place_queen(board, 0, 0, solutions)
return solutions
def place_queen(board, row, column, solutions):
"""place a queen that satisfies all the conditions"""
#base case
if row > len(board)-1:
#print board
solutions.append(deepcopy(board)) #Q1
#check every column of the current row if its safe to place a queen
while column < len(board):
if is_safe(board, row, column):
#place a queen
board[row][column] = 1
#place the next queen with an updated board
return place_queen(board, row+1, 0, solutions) #Q2
else:
column += 1
#there is no column that satisfies the conditions. Backtrack
for c in range(len(board)):
if board[row-1][c] == 1:
#remove this queen
board[row-1][c] = 0
#go back to the previous row and start from the last unchecked column
return place_queen(board, row-1, c+1, solutions) #Q3
O acima retorna todas as soluções encontradas em vez de imprimi-las. Eu tenho mais algumas perguntas (as partes relacionadas estão marcadas como Q1 e Q2 no código acima)
Por que precisamossolutions.append(deepcopy(board))
? Em outras palavras, o que exatamente está acontecendo quando fazemossolutions.append(board)
e por que isso leva a acrescentar a placa inicial que é[[0,0,0,0] ...]
?Por que nos deparamos com um problema quandoreturn place_queen(board, row+1, 0)
é substituído porplace_queen(board, row+1, 0)
? Nós não estamos realmente retornando nada (ouNone
), mas semreturn
a lista sai do índice.