一、题目链接
按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
n
皇后问题研究的是如何将 n
个皇后放置在
n
×
n
n \times n
n×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;
}
};