【LeetCode with Python】 N-Queens II

最新推荐文章于 2020-12-09 23:09:14 发布

3x3只眼

最新推荐文章于 2020-12-09 23:09:14 发布

阅读量1w

点赞数

本文链接：https://blog.youkuaiyun.com/nerv3x3/article/details/39453357

版权

LeetCode with Python 同时被 2 个专栏收录

104 篇文章

订阅专栏

LeetCode with Python

104 篇文章

订阅专栏

本文介绍了一种解决N皇后II问题的方法，该问题要求计算N皇后问题的所有不同解决方案的数量而非输出具体的棋盘配置。通过使用深度优先搜索算法，递归地检查每个可能的位置是否有效，并记录有效的解决方案数量。

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

博客域名： http://www.xnerv.wang
原题页面： https://oj.leetcode.com/problems/n-queens-ii/
题目类型：
难度评价：★
本文地址： http://blog.youkuaiyun.com/nerv3x3/article/details/39453357

Follow up for N-Queens problem.

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

class Solution:

    def __init__(self):
        self.paths_count = 0

    def dpTotalNQueens(self, n, prefix, level):
        for i in range(0, n):
            point = (i, level)
            is_valid = True
            for m in range(0, len(prefix)):
                if i == prefix[m]:
                    is_valid = False
                    break
                if level - m == i - prefix[m] or level - m == prefix[m] - i:
                    is_valid = False
                    break
            if is_valid:
                new_prefix = prefix[:]
                new_prefix.append(i)
                if n - 1 != level:
                    self.dpTotalNQueens(n, new_prefix, level + 1)
                else:
                    self.paths_count += 1

    # @return an integer
    def totalNQueens(self, n):
        self.dpTotalNQueens(n, [ ], 0)
        return self.paths_count