Leetcode 51. 皇后 回溯 C++实现

Leetcode 51. 皇后

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

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

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

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

算法：

创建二维返回数组 ans ，皇后所在位置数组 col ，列占用数组 on_path ,左斜线占用数组 diag1 ，右斜线占用数组 diag2 。

找到斜线占用的规律：

进入 dfs 函数，从第一行开始，进入递归，遍历当前行的每一列，找到满足条件的，有两种选择，选这个位置或者不选。选这个位置，存入 col 数组，递归下一行。不选，恢复现场，注意，已经存入数据的 col 可以不用 pop_back（），因为可以直接覆盖。

代码：

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> ans;// 返回数组ans
        vector<int> col(n),on_path(n),diag1(n*2-1),diag2(n*2-1);
        //行号r  列号c
        auto dfs = [&](auto &&dfs,int r){
            if(r == n){
                vector<string> board(n);
                for(int i = 0;i < n;i++)    board[i] = string(col[i],'.') + 'Q' + string(n - 1 - col[i],'.');
                ans.emplace_back(board);
                return ;
            }
            for(int c = 0;c < n;c++){
                int rc = r - c + n - 1;
                if(!on_path[c] && !diag1[r+c] && !diag2[rc]){
                    col[r] = c;// col存储皇后所在位置
                    on_path[c] = diag1[r+c] = diag2[rc] = true;
                    dfs(dfs,r + 1);// 进入下一行
                    on_path[c] = diag1[r+c] = diag2[rc] = false;// 恢复现场
                }
            }
        };
        dfs(dfs,0);// 递归入口
        return ans;
    }
};