55 leetcode - N-Queens II

最新推荐文章于 2025-08-08 00:00:33 发布

忧桑的小兔子

最新推荐文章于 2025-08-08 00:00:33 发布

阅读量471

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode N-Queens Queens N-Queens个数八皇后

本文链接：https://blog.youkuaiyun.com/lis_12/article/details/53840866

算法专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一个N皇后问题的变形，不输出具体的棋盘配置而是返回所有可能的解决方案总数。通过递归检查每一步是否符合N皇后规则，最终统计出所有可行解的数量。

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

N-Queens的变形.

#!/usr/bin/python
# -*- coding: utf-8 -*-
'''
N-Queens II
Now, instead outputting board configurations, return the total number of distinct solutions.
'''
class Solution(object):
    def __check(self,result,start_index,next_val):
        for index,val in enumerate(result[:start_index]):
            if (val == next_val) or (abs(index - start_index) == abs(val - next_val)): #将不符合N皇后规则的去掉
                return False

        return True

    def __solveNQueens(self,n,result,start_index,ret):
        if start_index == n:
            ret[0] += 1
        else:
            #index代表行数,val代表列数
            for i in range(n):
                if self.__check(result,start_index,i) == True:
                    result[start_index] = i
                    self.__solveNQueens(n,result,start_index + 1,ret)

    def totalNQueens(self, n):
        """
        :type n: int
        :rtype: List[List[str]]
        """
        if n == 1:
            return 1

        if n < 4:
            return 0

        ret = [0]
        self.__solveNQueens(n,[0]*n,0,ret)
        return ret[0]

if __name__ == "__main__":
    s = Solution()
    print s.totalNQueens(8)