N皇后
N皇后问题要求同一行、同一列还有斜对角线不能同时出现两个及以上的皇后。
思路:满足题目的三个需求,需要对棋盘的深度进行递归,递归的过程中进行回溯,也就是撤销上一次的移动。
class Solution {
private:
vector<vector<string>> ans;
void slove(int n,int r, vector<string> &temp){
if(r == n){
ans.push_back(temp);
return;
}
for(int c = 0;c<n;c++){
if(isValid(n,r,c,temp)){
temp[r][c] = 'Q';
slove(n,r+1,temp);
//回溯
temp[r][c] = '.';
}
}
}
bool isValid(int n,int r,int c,vector<string> &path){
//检查同一列
for(int i =0;i<r;i++){
if(path[i][c] =='Q')
return false;
}
//判断两条对角线
for(int i = r-1,j=c-1;i>=0&&j>=0; i--,j--){
if(path[i][j] == 'Q')
return false;
}
for(int i =r-1,j=c+1;i>=0&&j<n;i--,j++){
if(path[i][j] == 'Q')
return false;
}
return true;
}
public:
vector<vector<string>> solveNQueens(int n) {
ans.clear();
vector<string> temp(n, string(n, '.'));
slove(n,0,temp);
return ans;
}
};
看完之后,你是不是发现了代码中没有对棋盘的同一行进行判断,这是因为每一层递归只会选for循环 当前行 的一个元素。