力扣 51. N皇后

一、题目链接

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

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

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

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

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

示例 2：
输入：n = 1
输出：[["Q"]]

---

二、题解

class Solution {
private:
    vector<vector<string>> result;

public:
    vector<vector<string>> solveNQueens(int n) {
        vector<string> each_case(n, string(n, '.'));
        backtracking(each_case, 0);
        return result;
    }

    void backtracking(vector<string> &each_case, int row) {
        if (row == each_case.size()) {
            result.emplace_back(each_case);
            return;
        }

        for (int col = 0; col < each_case.size(); col++) {
            if (isValid(each_case, row, col)) {
                each_case.at(row).at(col) = 'Q';
                backtracking(each_case, row + 1);
                each_case.at(row).at(col) = '.';
            }
        }
    }

    bool isValid(vector<string> &each_case, int row, int col) {
        for (int i = 0; i < row; i++) {
            if (each_case.at(i).at(col) == 'Q') {
                return false;
            }
        }

        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
            if (each_case.at(i).at(j) == 'Q') {
                return false;
            }
        }

        for (int i = row - 1, j = col + 1; i >= 0 && j < each_case.size(); i--, j++) {
            if (each_case.at(i).at(j) == 'Q') {
                return false;
            }
        }

        return true;
    }
};