Python描述 LeetCode 52. N皇后 II

Posted 2022-06-16 亓官劼

Python描述 LeetCode 52. N皇后 II

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、CSDN、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息.

---

本文原创为【亓官劼】（qí guān jié ），请大家支持原创，部分平台一直在恶意盗取博主的文章！！！ 全部文章请关注微信公众号【亓官劼】。

题目

n 皇后问题 研究的是如何将 n 个皇后放置在 n × n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回 n 皇后问题 不同的解决方案的数量。

示例 1：

输入：n = 4
输出：2
解释：如上图所示，4 皇后问题存在两个不同的解法。

示例 2：

输入：n = 1
输出：1

提示：

• 1 <= n <= 9

Python描述

51题结果加个len即可

class Solution:
    def totalNQueens(self, n: int) -> int:
        ds = [['.' for _ in range(n)] for __ in range(n)]
        res = []
        ld =   # 行
        cd =   # 列
        zd =   # 主对角线
        fd =   # 副对角线

        def dfs(i, j):
            nonlocal n
            if i == n:
                if sum([item.count('Q') for item in ds]) == n:
                    res.append([''.join(item) for item in ds])
                return
            ni, nj = i if (j + 1) < n else i + 1, (j + 1) if (j + 1) < n else 0
            dfs(ni, nj)
            if ds[i][j] != 'Q' and ld.get(i, False) == cd.get(j, False) == zd.get(i + j, False) == fd.get(i - j,False) == False:
                ds[i][j] = 'Q'
                ld[i] = cd[j] = zd[i + j] = fd[i - j] = True
                dfs(ni, nj)
                ds[i][j] = '.'
                ld[i] = cd[j] = zd[i + j] = fd[i - j] = False

        dfs(0, 0)
        return len(res)