【回溯法】51. 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 <= n <= 9
皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

思路：回溯法

一行一行查找皇后的放置位置, 回溯的过程中标记皇后所在两条对角线以及列
红色的对角线 k1=y-x ，数组下标可能是负数，所以要加上偏移量n ，数组下标不能是负数。
关于对角线的标记:

class Solution {
public:
    vector<vector<string>>ans;
    vector<string>path;
    vector<bool>col,dg,udg;   //列，两条对角线
    int n;

    vector<vector<string>> solveNQueens(int _n) {
        n=_n;
        col=vector<bool>(n);
        dg=udg=vector<bool>(n*2);
        path=vector<string>(n,string(n,'.'));

        dfs(0);
        return ans;
    }

    void dfs(int u){
        if(u==n){
            ans.push_back(path);
            return;
        }
        //枚举第u行皇后可以放到哪个位置
        for(int i=0;i<n;i++){
            if(!col[i] && !dg[u-i+n] && !udg[u+i]){
                col[i]=dg[u-i+n]=udg[u+i]=true;
                path[u][i]='Q';
                dfs(u+1);
                col[i]=dg[u-i+n]=udg[u+i]=false;
                path[u][i]='.';

            }
        }
    }
};