python leetcode 52. N-Queens II

Neekity

于 2018-12-08 17:28:05 发布

阅读量572

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/Neekity/article/details/84894893

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文介绍了一种基于深度优先搜索(DFS)思想的N皇后问题解决方案，该方案通过统计有效放置皇后的方法数量，避免了列出所有可能的棋盘布局，从而提高了计算效率。

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

还是DFS的思想，只不过不需要列出所有情况，只需要统计次数。

当然你可以在 N-Queens的基础上return len（res）

class Solution:
    def totalNQueens(self, n):
        """
        :type n: int
        :rtype: List[List[str]]
        """
        def check(row,num):
            for i in range(row):
                if board[i]==num or abs(num-board[i])==abs(i-row):
                    return False 
            return True 
        board=[-1]*n         
        count=row=num=0
        while row<n:
            while num<n:
                if check(row,num):
                    board[row]=num 
                    num=0
                    break 
                else:
                    num+=1
            if board[row]==-1:#这行找不到有效放置方法，退回上一层
                if row==0: #退出循环
                    break
                row-=1
                num=board[row]+1
                board[row]=-1
                continue
            if row==n-1:
                count+=1
                num=board[row]+1
                board[row]=-1
                continue  
            row+=1
        return count