题目大意:将N个皇后摆放在N*N的棋盘上,要求N个皇后彼此间不能相互攻击。
算法思想:
采用一个数组C[n]记录每一行皇后所在位置的列下标,一个数组column[n]记录当前局面以后皇后所在的列,然后用两个数组main_dialog[2*n]和anti_dialog[2*n]标记皇后所在位置的主副对角线(通过规律可知,同一主对角线上的坐标其横纵坐标差值恒等,同一副对角线上的坐标横纵坐标和值恒等)。然后采用深搜,对于每一行来说搜索皇后所在列的所有可行方案,当所在列使得主对角线不冲突,行列不冲突则是一种可行方案,然后进行下一层的递归搜索。
代码如下:
class Solution {
public:
vector<vector<string> > solveNQueens(int n) {
this->column=vector<int>(n,0);//标记已存在的皇后的列
this->main_dialog=vector<int>(n*2,0);//标记主对角线
this->anti_dialog=vector<int>(n*2,0);//标记副对角线
vector<vector<string> > res;
vector<int> C(n,0);//记录每行皇后所在的列下标
dfs(C,res,0);
return res;
}
void dfs(vector<int> &C,vector<vector<string> >&res,int row){
const int N=C.size();
if(N==row){
vector<string> solution;
for(int i=0;i<N;++i){//行
string s(N,'.');
for(int j=0;j<N;++j){//列
if(C[i]==j) s[j]='Q';
}
solution.push_back(s);
}
res.push_back(solution);
return ;
}
for(int j=0;j<N;++j){
bool ok=column[j]==0&&main_dialog[row-j+N]==0&&anti_dialog[row+j]==0;
if(!ok) continue;
C[row]=j;
main_dialog[row-j+N]=anti_dialog[row+j]=column[j]=1;
dfs(C,res,row+1);
C[row]=0;
main_dialog[row-j+N]=anti_dialog[row+j]=column[j]=0;
}
}
protected:
vector<int> column;
vector<int> main_dialog;
vector<int> anti_dialog;
};