51. N 皇后

在这里插入图片描述
在这里插入图片描述
和数独差不多,要注意的是,有些操作能省则省,可以省下不少时间。自己定义一个二维数组存n*n的棋盘会很方便。
检测当前能否填皇后的函数是要检查同一行同一列,以及左上45度以及右上45度,因为我们是从上往下填的,所以只需要判断上方的就行了。
如果到了递归边界,就通过该bord生成一个vector然后加入res,这样不用自己手动维护vector< string >了很方便。
dfs中就是遍历当前行的每一列看看能否填上Q,如果可以就填上,然后直接去下一行,这就是我说的省下来的操作,不用去下一列了,当前行已经不可能再填皇后了,保持原有的.就行了。

class Solution {
public:
    bool isValid(int row, int col ,vector<vector<char>>& board){
        int n = board.size();
        for(int i = 0; i < n; ++i){
            if(board[i][col] == 'Q') return false;
        }
        for(int i = row, j = col; i >= 0 && j >= 0; --i,--j){
            if(board[i][j] == 'Q') return false;
        }
        for(int i = row, j = col; i >= 0 && j < n; --i, ++j){
            if(board[i][j] == 'Q') return false;
        }
        return true;
    }
    vector<string> trans(vector<vector<char>>& board){
        vector<string> v;
        for(int i = 0; i < board.size(); ++i){
            string x = "";
            for(int j = 0; j < board[0].size(); ++j){
                x += board[i][j];
            }
            v.push_back(x);
        }
        return v;
    }
    void dfs(int row, int col, vector<vector<char>>& board){
        if(row == board.size()){
            //这个时候将board里的拿出来作为一个vector返回
            vector<string> t = trans(board);
            res.push_back(t);
            return;
        }
        //如果当前行到最后一列也不满足,直接return
        for(int i = 0; i < board.size(); ++i){
            if(isValid(row,i,board)){
                board[row][i] = 'Q';
                dfs(row+1,0,board);
                board[row][i] = '.';
            }
        }
    }
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<char>> board(n,vector<char>(n,'.'));
        dfs(0,0,board);
        return res;
    }
private:
    vector<vector<string>> res;
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值