leetcode_52. N皇后 II

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

上图为 8 皇后问题的一种解法。

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

示例:

输入: 4
输出: 2
解释: 4 皇后问题存在如下两个不同的解法。
[
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],

["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]

class Solution {
public:
    int output=0;
    
    int totalNQueens(int n) {
        vector<string> res(n,string(n,'.'));
        putone(res,0);
        return output;
    }
    
    void putone(vector<string> &res,int i){
        int t = res.size();
        if(i>=t){
            ++output;
            return;
        }
        for(int j=0;j<t;j++)
        {
            if(canput(res,i,j))
            {
                res[i][j]='Q';
                putone(res,i+1);
                res[i][j]='.';
            }
        }
    }
    

    bool canput(vector<string>& board,int row,int col){
        for(int i=0;i<row;++i){
            if(board[i][col]=='Q') return false;
        }
        for(int r=row-1,c=col-1;r>=0&&c>=0;--r,--c){
            if(board[r][c]=='Q') return false;
        }
        for(int r=row-1,c=col+1;r>=0&&c<board[0].size();--r,++c){
            if(board[r][c]=='Q') return false;
        }
        return true;
    }
};