The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.
Given an integer n, return all distinct solutions to the n-queens puzzle.
Each solution contains a distinct board configuration of the n-queens’ placement, where ‘Q’ and ‘.’ both indicate a queen and an empty space respectively.
Example:
Input: 4
Output:
[
[".Q..", // Solution 1
"...Q",
"Q...",
"..Q."],
["..Q.", // Solution 2
"Q...",
"...Q",
".Q.."]
]
Explanation: There exist two distinct solutions to the 4-queens puzzle as shown above.
思路: 皇后之间不能互相攻击的前提是,皇后不能同在一条直线上(包括横竖两个方向)和皇后不能再同一对角线上。
根据这个规则,每一行只能摆放一个皇后,对这个皇后尝试所有可能的列(从第1-第n列,n为棋盘大小),从第一行开始,对每一行皇后的位置尝试所有可能,逐行开始摆放皇后,如果遇到某一行皇后在任何位置摆放都会与之前摆放的皇后冲突,那么回溯到前一步,修改上一行皇后的位置,若还不行则依次回溯。
代码
class Solution {
public:
vector<vector<string>> ans;//存放最后的答案
vector<vector<string>> solveNQueens(int n) {
vector<int> position(n+1, 0);//存放皇后摆放的位置,第i个位置上的数值表示第i行皇后所在的列
if (n!=0)
NQueens(position, n, 1);//从棋盘第一行开始尝试摆放皇后
return ans;
}
void NQueens(vector<int>& position, int n, int k)//n个皇后需要摆放,假设前k-1行皇后已经摆放妥当,该函数尝试摆放第k行的皇后
{
if(k == n+1)
{
vector<string> rst(n, string(n, '.'));
for (int i = 1; i<=n; i++)
{
rst[i-1][position[i]-1] = 'Q';
}
ans.push_back(rst);
return;
}
int i, j;
for (i = 1; i <= n; i++)//第k行尝试所有位置[1-n]
{
for(j = 1; j<k; j++)//遍历position,找到第k行皇后的合适位置
{
if (position[j]==i || abs(j-k)==abs(position[j]-i))//第k行尝试的位置与之前已经放好的皇后冲突
break;//尝试下一种摆放方法
}
if (j == k)//上面的for循环正常结束,没有冲突
{
position[k] = i;//摆放第k行的皇后
NQueens(position, n, k+1);
}
}
}
};