Solve N queens
After a five year hiatus, I decided to attempt to solve the famous N queens problem again. This time, instead of modeling the chess board using a `[[Bool]]`, I'm using `[Integer]` where the `Integer` indicates which column has a queen. This is a bit lighter in RAM.
This commit is contained in:
parent
14f6169fcf
commit
7672049e1c
1 changed files with 46 additions and 0 deletions
46
scratch/facebook/n-queens.py
Normal file
46
scratch/facebook/n-queens.py
Normal file
|
@ -0,0 +1,46 @@
|
||||||
|
def print_board(board):
|
||||||
|
result = []
|
||||||
|
for row in range(8):
|
||||||
|
r = []
|
||||||
|
for col in range(8):
|
||||||
|
r.append("X" if col == board[row] else "-")
|
||||||
|
result.append(" ".join(r))
|
||||||
|
print("\n".join(result))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def can_place(board, row, col):
|
||||||
|
column_occupied = not any([board[i] == col for i in range(row)])
|
||||||
|
|
||||||
|
diagonals_clear = True
|
||||||
|
for r in range(row):
|
||||||
|
w = abs(col - board[r])
|
||||||
|
h = abs(r - row)
|
||||||
|
if w == h:
|
||||||
|
diagonals_clear = False
|
||||||
|
break
|
||||||
|
|
||||||
|
return all([column_occupied, diagonals_clear])
|
||||||
|
|
||||||
|
def init_board():
|
||||||
|
board = []
|
||||||
|
for row in range(8):
|
||||||
|
board.append(None)
|
||||||
|
return board
|
||||||
|
|
||||||
|
def copy_board(board):
|
||||||
|
return board[:]
|
||||||
|
|
||||||
|
def n_queens():
|
||||||
|
do_n_queens(init_board(), 0, 0)
|
||||||
|
|
||||||
|
def do_n_queens(board, row, col):
|
||||||
|
if row == 8:
|
||||||
|
print_board(board)
|
||||||
|
return
|
||||||
|
for i in range(col, 8):
|
||||||
|
if can_place(board, row, i):
|
||||||
|
copy = copy_board(board)
|
||||||
|
copy[row] = i
|
||||||
|
do_n_queens(copy, row + 1, 0)
|
||||||
|
|
||||||
|
n_queens()
|
Loading…
Reference in a new issue