经典回溯,以一个棋盘为例,棋盘的一行代表决策树的一个层次,棋盘某一行上的一列代表一个选择,做出选择,进入递归,撤销选择,这就是回溯的模板。正如labuladong大佬总结的:
递归结束的条件就是做出的选择达到决策树的叶子,将路径计入结果。选择列表中做选择,一系列选择构成路径。
class Solution {
bool isValid(vector<string>& track, int row, int col, int n){
//同一列上没有
for(int i=row-1;i>=0;i--){
if(track[i][col] == 'Q')
return false;
}
//右上方没有
//ij的判断条件注意写成&&而不是逗号表达式,逗号表达式会导致vector访问越界
for(int i=row-1,j=col+1;i>=0 && j<n;i--,j++){
if(track[i][j] == 'Q')
return false;
}
//左上方没有
for(int i=row-1,j=col-1;i>=0 && j>=0;i--,j--){
if(track[i][j] == 'Q')
return false;
}
//合法的前提是有三个可能不合法的条件
return true;
}
void followTrack(vector<vector<string>>& res, vector<string>& track, int row, int n){
if(row == n){
res.push_back(track);
return ;
}
//枚举这一行每一列的位置上的情况
for(int i=0; i<n; i++){
if(!isValid(track, row, i, n))
continue;
track[row][i]='Q';
followTrack(res, track, row+1, n);
track[row][i]='.';
}
}
public:
vector<vector<string>> solveNQueens(int n) {
vector<vector<string>> res;
vector<string> track(n, string(n,'.'));
followTrack(res, track, 0, n);
return res;
}
};