n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
上图为 8 皇后问题的一种解法。
给定一个整数 n,返回所有不同的 n 皇后问题的解决方案。
每一种解法包含一个明确的 n 皇后问题的棋子放置方案,该方案中 'Q'
和 '.'
分别代表了皇后和空位。
示例:
输入: 4 输出: [ [".Q..", // 解法 1 "...Q", "Q...", "..Q."], ["..Q.", // 解法 2 "Q...", "...Q", ".Q.."] ] 解释: 4 皇后问题存在两个不同的解法。
思路:写代码很有意思,回头再补充。
C++代码如下:
class Solution {
vector<vector<string>> res;
vector<bool> col,dia1,dia2; //列,两个对角线是否存在
//n代表的是处理n皇后问题,摆放第index行皇后的位置 row为已经摆放的皇后
void putQueen(int n,int index,vector<int>& row)
{
if( index == n){
res.push_back(generate(n,row));
return;
}
for(int i=0; i<n;i ++)
{
//尝试将第index皇后摆放在第i列
if( !col[i] && !dia1[index+i] && !dia2[index-i+n-1])
{
row.push_back(i);
col[i]=true;
dia1[index+i]=true;
dia2[index-i+n-1]=true;
putQueen(n,index+1,row);
//回溯
col[i]=false;
dia1[index+i]=false;
dia2[index-i+n-1]=false;
row.pop_back();
}
}
return;
}
vector<string> generate(int n,vector<int>& row)
{
vector<string> res(n,string(n,'.'));
for(int i=0;i<row.size();i++)
{
res[i][row[i]]='Q';
}
return res;
}
public:
vector<vector<string>> solveNQueens(int n) {
vector<int> row;
col=vector<bool>(n,false);
dia1=vector<bool>(n*2-1,false);
dia2=vector<bool>(n*2-1,false);
putQueen(n,0,row);
return res;
}
};