问题
按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
一、解决思路
寻找不同的放置方案,每个方案都是包含n个皇后在nxn矩阵的放置问题,再进一步细想,每个nxn矩阵的每一行放置都有着很明显的规律,做重复有规律的事情就会想到使用递归方法,有递归就会有回溯。
首先本题的规律就是皇后的约束条件:不能同行,不能同列,不能同斜线。
那么在二维矩阵中,高相当于树的高度,也就是递归的深度,宽相当于树的宽度,也就是每一层中可能会出现的结果。
因此不难写出递归回溯方案:
即满足条件的行加入到棋盘中,为什么是行呢?(其实列也可以),行代表的是深度,一行代表一个皇后,当行数即递归的深度等于n时即找到了一个方案,值得注意的是每次递归的行数都要从0开始,递归的逻辑很简单,满足条件则相应位置的皇后加入棋盘(二维矩阵),然后向下递归,每一次递归终止后,则得到了完整的棋盘,然后进行回溯,寻找别的方案。
另外一个难点在于编写约束条件:
不能同行,不能同列:则通过判断即将加入棋盘的行,是否与已经在棋盘的行同列即可,一个for循环遍历。
不能同斜线,这包括了两个对角线:需要判断对角线上是否有皇后即可。
二、代码
代码如下(示例):
class Solution {
private:
vector<vector<string>> result;
void backTracking(int n, int row, vector<string>& chessboard){
if (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;
}
}
//检查45°对角线
for (int i = row-1, j=col-1; i>=0 && j>=0; i--, j--){
if (chessboard[i][j]=='Q'){
return false;
}
}
//检查135°对角线
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;
}
};
总结
题目要点虽然不难,但是想清楚每个逻辑要求还是很高的。这道题主要考察了回溯,还有约束条件的判断,只要理解这是一个回溯过程和会写约束条件的代码就解决了。