这道题是一类比较经典的题目,和数独以及leetcode上的其他题目一起构成了比较典型的一个类型题集,本质就是迷宫问题,基本的解决算法就是深度优先遍历DFS。
题目的大致意思是:在一个棋盘上,每一行和每一列都有且只能有一个Q棋子,且45度对角线上也只能由一个Q棋子。
实际上我们就一行一行的放置,第一行选择一个位置放置Q,接着到第二行放置,因为深度遍历的时候已经考虑到了行的问题,本身就是按照行数进行深度优先遍历,因此,当我们打算在(i,j)位置上放置Q棋子,我们需要检查i行以前的,在j列上是否有Q棋子(i行之后还没有遍历到,不可能有Q),即(i-1,j),(i-2,j)……,此外还要考虑对角线的情况,即该点(i,j)的左上角和右上角两个对角线上是否有Q,如果没有,则该位可以置为Q,继续遍历。
AC代码:
class Solution {
public:
vector<vector<string>> solveNQueens(int n) {
vector<string> curres (n, string(n, '.'));
solveNQueens (curres, 0);
return res;
}
void solveNQueens(vector<string>& curres, int row) {
if (row == curres.size()) {
res.emplace_back(curres);
return ;
}
for (int i=0; i<curres.size(); ++i) {
if (isGood(curres, row, i)) {
curres[row][i] = 'Q';
solveNQueens(curres, row+1);
curres[row][i] = '.';
}
}
}
bool isGood (vector<string>& curres, int row, int col) {
for (int i=0; i<row; ++i) {
if (curres[i][col] == 'Q') {
return 0;
}
}
for (int i=row-1, j=col+1; i>=0 && j<curres[0].size(); --i, ++j) {
if (curres[i][j] == 'Q') {
return 0;
}
}
for (int i=row-1, j=col-1; i>=0 && j>=0; --i, --j) {
if (curres[i][j] == 'Q') {
return 0;
}
}
return 1;
}
vector<vector<string>> res;
};