n 皇后问题 研究的是如何将 n
个皇后放置在 n × n
的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n
,返回 n 皇后问题 不同的解决方案的数量。
输入: n = 4 输出: 2 解释: 如上图所示,4 皇后问题存在两个不同的解法。
输入: n = 1 输出: 1
1 <= n <= 9
class Solution:
def totalNQueens(self, n: int) -> int:
ret = 0
for ys in itertools.permutations(range(n)):
board = [[False] * n for _ in range(n)]
for x, y in zip(range(n), ys):
if any(board[x - i][y - i] for i in range(1, min(x, y) + 1)) or \
any(board[x - i][y + i] for i in range(1, min(x, n - y - 1) + 1)):
break
board[x][y] = True
else:
ret += 1
return ret