52. N-Queens II Leetcode Python

最新推荐文章于 2022-11-15 16:32:31 发布

hyperbolechi

最新推荐文章于 2022-11-15 16:32:31 发布

阅读量1.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode dfs python

本文链接：https://blog.youkuaiyun.com/hyperbolechi/article/details/43728767

leetcode 专栏收录该内容

166 篇文章

订阅专栏

本文介绍了一种解决N皇后问题的方法，通过深度优先搜索（DFS）算法来计算所有可能的不同解决方案的数量。使用了一个board数组来记录每行皇后的位置，并通过递归的方式检查每个放置位置的有效性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

这题的解法是建一个board表示行比如n=4 时候board=[1,3,0,2] 表示 0Q00,000Q,Q000,00Q0

利用对于dfs每次确认当前Board是否满足条件时候已经表示之前的格子都已经满足条件了。

在check 函数表示 queen 是否可以放在第k（depth）行j(row)列

class Solution:
    # @return an integer
    def __init__(self):
        self.count=0
    def check(self,k,j,board):
        for i in range(k):
            if board[i]==j or abs(k-i)==abs(board[i]-j):
                return False
        return True
    def dfs(self,depth,board,n):
        if depth==n:
            self.count+=1
        for row in range(n):
            if self.check(depth,row,board):
                s='.'*n
                board[depth]=row
                self.dfs(depth+1,board,n)
    def totalNQueens(self, n):
        board=[-1 for i in range(n)]
        self.dfs(0,board,n)
        return self.count