Fix n-queues problem

infrablue1 · infrablue1 · commit 267737745308 · 2025-02-18T20:37:30.000+08:00
diff --git a/backtracking/n_queens.py b/backtracking/n_queens.py
@@ -40,8 +40,7 @@ def is_safe(board: list[list[int]], row: int, column: int) -> bool:
         and all(
             board[i][j] != 1 for i, j in zip(range(row, -1, -1), range(column, -1, -1))
         )
-        and all(board[i][j] != 1 for i, j in zip(range(row, n), range(column, n)))
-        and all(board[i][j] != 1 for i, j in zip(range(row, n), range(column, -1, -1)))
+        and all(board[i][j] != 1 for i, j in zip(range(row, -1, -1), [column] * (row + 1)) )
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -40,8 +40,7 @@ def is_safe(board: list[list[int]], row: int, column: int) -> bool:`
`40`	`40`	`and all(`
`41`	`41`	`board[i][j] != 1 for i, j in zip(range(row, -1, -1), range(column, -1, -1))`
`42`	`42`	`)`
`43`		`- and all(board[i][j] != 1 for i, j in zip(range(row, n), range(column, n)))`
`44`		`- and all(board[i][j] != 1 for i, j in zip(range(row, n), range(column, -1, -1)))`
	`43`	`+ and all(board[i][j] != 1 for i, j in zip(range(row, -1, -1), [column] * (row + 1)) )`
`45`	`44`	`)`
`46`	`45`
`47`	`46`