LeetCode51：Python--N皇后问题

最新推荐文章于 2024-04-26 09:10:56 发布

原创

最新推荐文章于 2024-04-26 09:10:56 发布 · 392 阅读

0 ·

CC 4.0 BY-SA版权

本文作者分享了自己对于LeetCode51题——N皇后问题的理解，认为已有的解法阐述不清，于是提供了自己清晰的解决方案。如果需求是求解的数量，可以通过全局变量在遍历结束后记录，或者直接返回解的数组长度。

看了很多人讲述的题解，都觉得不够清楚。然后自己写了一下，应该很清楚吧

如果是返回解的个数，那就直接用一个全局变量在遍历退出的地方记录一下就好了

或者返回len(ans)，哈哈哈

class Solution(object):
    """
    @param: n: The number of queens
    @return: All distinct solutions
    """
    def solveNQueens(self, n):
        # write your code here
        ans = []
        
        def dfs(nums ,row):
            #从上到下一行一行放皇后
            #nums[row] = col,表示第row行的皇后放在第col列
            if row == n:
                #如果放到最后一行，表示放完了
                ans.append(nums[:])
                return 
            for col in range(n):
                #对于此时的第row行，我们从左往右先尝试着放皇后，nums[row] = col
                nums[row] = col
                if vaild(nums,row):
                    #如果此时放在row行的第col列，这个位置时是合法的，
                    #ok继续向下一行row+1行放皇后
                    dfs(nums,row + 1)
                    
        def vaild(nums,row):
            #对于nums[0:row-1]，已经是合理的放置位置。
            #那么此时放在row行的第nums[row]位置，是否是合法的
            for i in