LeetCode - 解题笔记 - 51 - N-Queens

最新推荐文章于 2022-03-25 16:53:26 发布

支锦铭

最新推荐文章于 2022-03-25 16:53:26 发布

阅读量133

点赞数

分类专栏： LeetCode解题笔记文章标签： leetcode dfs 算法

本文链接：https://blog.csdn.net/cary_leo/article/details/115880444

版权

N-Queens

Solution 1

【这个题虽然是模版题但是我还是忘了最经典的部分】首先声明这是一道dfs模版题，其次说明我看了题解😭。这道题的DFS并不是难点，直接暴力实现也不是不行，但是可以进一步优化其所用的空间（就是这个部分，当时学算法的时候就学习了这个优化方法，很不负众望地忘记了！）。

如果使用传统的DFS的思路，就是创建一个二维的状态棋盘，从第一行到最后一行逐行的各个位置便利，同时记录所在的列和两条对角线的禁止状态。

当然这样写没有任何问题（其实能过），但是这里的状态棋盘可以进一步优化：使用三个一维数组分别记录当前禁止的列和对角线。列比较好找，两条对角线需要额外计算一个斜率（就这里我忘记了！看的题解！对自己很生气！）。对于左下到右上的对角线，斜率是行号加列号；另外一条就是行号减列号。因此记录对角线状态的数组会长一倍（ $2 n$ ）。

时间复杂度： $O (n!)$ ，其中 $n$ 为放置的皇后个数，对应棋盘的尺寸。
空间复杂度： $O (n)$ ，其中 $n$ 为放置的皇后个数，对应棋盘的尺寸。

class Solution {
   
public:
    vector<vector<string>> solveNQueens(int n) {
   
        vector<vector<string>> ans = vector<vector<string>>();
        vector<int> posRow = vector<int>(n, -1); // 每行皇后的位置
        vector<bool> stateCol = vector