Skip to content

Latest commit

 

History

History
42 lines (34 loc) · 1.14 KB

README_CN.md

File metadata and controls

42 lines (34 loc) · 1.14 KB

52. N皇后 II

n 皇后问题 研究的是如何将 n 个皇后放置在 n × n 的棋盘上,并且使皇后彼此之间不能相互攻击。

给你一个整数 n ,返回 n 皇后问题 不同的解决方案的数量。

示例 1:

输入: n = 4
输出: 2
解释: 如上图所示,4 皇后问题存在两个不同的解法。

示例 2:

输入: n = 1
输出: 1

提示:

  • 1 <= n <= 9

题解 (Python)

1. 题解

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