题目(leecode T51):
按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
n 皇后问题 研究的是如何将 n
个皇后放置在 n×n
的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n
,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q'
和 '.'
分别代表了皇后和空位。
方法:本题是回溯算法解决的经典问题之---N皇后问题,问题主要是解决皇后在棋盘中的约束问题,即皇后不能同行同列同斜线,这个约束可以通过一个简单的判断函数来辅助实现,主要是如何遍历N皇后的位置,是需要考虑的,其实N皇后问题也可以抽象成一个树形结构来实现,树中的每一行是对矩阵的每一行的遍历,每行中皇后有N个可能的位置,同时对这N个位置进行判断是否可行。分析回溯三部曲:
1:传入参数与返回值:传入棋盘大小N,矩阵行数row和棋盘状态chessborad二维数组
2:终止条件:终止是当我们已经在每一行找到了一个适合放皇后的位置,因此就是我们遍历的行数row等于N时
3:单层处理逻辑:每一层中,我们需要对每行中的每一列做判断,判断该位置是否能放皇后Q,如果可以就将该位置设置为Q,不可以就跳过。然后递归和回溯。
题解:
class Solution {
private:
vector<vector<string>> result;
void backtracking(int n, int row, vector<string>& chessboard){
if(row == n){ //终止条件,当遍历行数row等于棋盘大小N
result.push_back(chessboard);
return;
}
for(int col = 0; col < n; col++){ //行数是固定的,根据列数遍历
if(isValid(row, col, chessboard, n)){ //判断当前位置能否放置皇后
chessboard[row][col] = 'Q';
backtracking(n ,row + 1, chessboard);
chessboard[row][col] = '.';
}
}
}
bool isValid(int row, int col, vector<string>&chessboard, int n){ //判断同列有无皇后
for(int i = 0; i < row; i++){
if(chessboard[i][col] == 'Q'){
return false;
}
}
for(int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--){ //检查左斜线
if(chessboard[i][j] == 'Q'){
return false;
}
}
for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++){ //检查右斜线
if(chessboard[i][j] == 'Q'){
return false;
}
}
return true;
}
public:
vector<vector<string>> solveNQueens(int n) {
result.clear();
std::vector<std::string> chessboard(n , std::string(n, '.')); //初始化棋盘
backtracking(n , 0, chessboard);
return result;
}
};