LeetCode 51.N皇后
n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击(使其中任意两个皇后都不同列、同行和在一条斜线上 )。
给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
class Solution {
public:
vector<vector<string>> solveNQueens(int n) {
vector<string> path(n,string(n,'.')); // 初始化path
backtracking(n, 0, path); // 从第1行开始回溯
return result;
}
private:
vector<vector<string>> result;
void backtracking(int n, int row, vector<string>& path) {
if (row == n) {
result.push_back(path);
return;
}
for (int col = 0; col < n; col++) { // 遍历列
if (isValid(row, col, path, n)) { // 验证合法就可以放
path[row][col] = 'Q'; // 放置皇后
backtracking(n, row + 1, path);
path[row][col] = '.'; // 回溯,撤销皇后
}
}
}
bool isValid(int row, int col, vector<string>& path, int n) {
//int count = 0;
// 检查列
for (int i = 0; i < row; i++) { // 这是一个剪枝
if (path[i][col] == 'Q') {
return false;
}
}
// 检查 135度角(左上)是否有皇后
for (int i = row - 1, j = col - 1; i >=0 && j >= 0; i--, j--) {
if (path[i][j] == 'Q') {
return false;
}
}
// 检查 45度角(右上)是否有皇后
for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
if (path[i][j] == 'Q') {
return false;
}
}
return true;
}
};