原题地址:https://leetcode-cn.com/problems/n-queens-ii/description/
题目描述:
n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
上图为 8 皇后问题的一种解法。
给定一个整数 n,返回 n 皇后不同的解决方案的数量。
示例:
输入: 4 输出: 2 解释: 4 皇后问题存在如下两个不同的解法。 [ [".Q..", // 解法 1 "...Q", "Q...", "..Q."], ["..Q.", // 解法 2 "Q...", "...Q", ".Q.."] ]
解题思路:
和上题思路一样。。
class Solution {
public:
bool isvalid(vector<string> &temp, int i, int j){//判断棋盘是否有效
//for (int k = 0; k<temp[i].size(); ++k){//判断行。不用判断行了,每行放一个之后就会递归到下一行了
// if (temp[i][k] == 'Q') return false;
//}
for (int k = 0; k < i; ++k){//判断列
if (temp[k][j] == 'Q') return false;
}
for (int p = i - 1, q = j - 1; p>=0 && q>=0; --p, --q){//判断左上对角线
if (temp[p][q] == 'Q') return false;
}
for (int p = i - 1, q = j + 1; p >= 0 && q<temp.size(); --p, ++q){//判断右上对角线
if (temp[p][q] == 'Q') return false;
}
return true;
}
int dfs(int &count, vector<string> &temp, int i, int n){
if (i == n)
return ++count;
for (int j = 0; j<n; ++j){
if (isvalid(temp, i, j)) {
temp[i][j] = 'Q';//递归前修改
dfs(count, temp, i + 1, n);
}
temp[i][j]='.';//递归后恢复
}
return count;
}
int totalNQueens(int n) {
int count = 0;
string aa;
for (int i = 0; i < n; ++i)
aa += '.';
vector<string>temp(n,aa);
return dfs(count, temp, 0, n);
}
};