algorithm/hello-algo
1
0
Fork 0
mirror of https://github.com/krahets/hello-algo.git synced 2025-12-19 07:17:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update n_queens code.

Browse Source
This commit is contained in:
krahets
2023-06-21 02:56:28 +08:00
parent ba1e5c1d7d
commit 9fc1a0b2b3
8 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
codes/python/chapter_backtracking/n_queens.py
View File

@@ -24,8 +24,8 @@ def backtrack(
# 计算该格子对应的主对角线和副对角线
diag1 = row - col + n - 1
diag2 = row + col
# 剪枝：不允许该格子所在 (列 或 主对角线 或 副对角线) 包含皇后
if not (cols[col] or diags1[diag1] or diags2[diag2]):
# 剪枝：不允许该格子所在列、主对角线、副对角线存在皇后
if not cols[col] and not diags1[diag1] and not diags2[diag2]:
# 尝试：将皇后放置在该格子
state[row][col] = "Q"
cols[col] = diags1[diag1] = diags2[diag2] = True