005-4算法笔记【回溯】N皇后

51. N 皇后

题目

按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

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

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

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例 1：

输入：n = 4
输出：[[“.Q…”,“…Q”,“Q…”,“…Q.”],[“…Q.”,“Q…”,“…Q”,“.Q…”]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

示例 2：

输入：n = 1
输出：[[“Q”]]

分析

1. 解向量：二维数组chessboard[N][N]
2. 解空间：n叉排列树
3. 逐行生成棋盘，每一行有且只有一个Queen，有n种可能情况，排列树的每一个节点有n个子节点；
4. 每个节点检查是否合法，即每行、每列、两个对角线都只能有一个Queen
5. 复杂度：O(n!)，排列树有n!个节点，每个节点检查合法性n²，T(n)=O(n² * n!)=O(n!)

class Solution {
    vector<vector<string>> res;
    //回溯:生成第idx行皇后的位置
    void backtrack(int idx,int n,vector<string>chessboard){
        if(idx==n){
            res.push_back(chessboard);
            return;
        }
        for(int j=0;j<n;j++){
            if(isValid(chessboard,n,idx,j)){//chessboard[row][col]可以放Q
                chessboard[idx][j]='Q';
                backtrack(idx+1,n,chessboard);
                //恢复
                chessboard[idx][j]='.';
            }
        }
    }
    //检查chessboard[row][col]是否可以放Q
    bool isValid(vector<string>& chessboard,int n,int row,int col){
        //检查列:[0:row-1][col]是否已经有皇后
        for(int i=0;i<row;i++){
            if(chessboard[i][col]=='Q')
                return false;
        }
        //检查45°斜线
        for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--){
            if(chessboard[i][j]=='Q')
                return false;
        }
        //检查135°斜线
        for(int i=row-1,j=col+1;i>=0&&j<n;i--,j++ ){
            if(chessboard[i][j]=='Q')
                return false;
        }
        return true;
    }
public:
    vector<vector<string>> solveNQueens(int n) {
        res.clear();
        //初始：没有放置皇后，全是"."
        vector<string>chessboard(n,string(n,'.'));
        backtrack(0,n,chessboard);
        return res;
    }
};