n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
上图为 8 皇后问题的一种解法。
给定一个整数 n,返回 n 皇后不同的解决方案的数量。
示例:
输入: 4
输出: 2
解释: 4 皇后问题存在如下两个不同的解法。
[
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],
["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]
解题思路:
与之前写的N皇后问题一致,最后输出改为个数即可。
C++代码:
class Solution {
public:
int totalNQueens(int n) {
if (n <= 0)return 1;
if (n == 1)return 1;
N = n;
Queens = vector(n, 0);//N个皇后的位置都在每一行的第一个
DFS(0);
return res.size();
}
private:
int N;
vector<vector>res;
vectorQueens;
private:
bool isOk(pair<int, int>a, pair<int, int>b)//判断位置是否相同
{
if (a.first == b.first)return false;
if (a.second == b.second)return false;
if (abs(a.first - b.first) == abs(a.second - b.second))return false;
return true;
}
string int2str(int pos, int N)
{
return string(pos, ‘.’) + string(1, ‘Q’) + string(N - (pos + 1), ‘.’);
}
void DFS(int num)
{
int i, j;
pair<int, int>a, b;
if (num == 0)
{
for (int i = 0; i < N; i++)
{
DFS(num + 1);
Queens[num]++;//第num个皇后的位置,从1开始,需加一
}
}
if (num>0 && num < N - 1)
{//找到所有与前面皇后兼容的点,没有就return
for (int i = 0; i < N; ++i)
{
Queens[num] = i;
a = pair<int, int>(num, Queens[num]);
//判断第num个皇后与前面皇后的兼容问题
bool sgn = true;
for (j = 0; j < num; j++)
{
b = pair<int, int>(j, Queens[j]);
if (!isOk(a, b)){ sgn = false; break; }
}
if (sgn == true)DFS(num + 1);
}
}
if (num == N - 1)//到达最后一个皇后
{
for (int i = 0; i < N; ++i)
{
Queens[num] = i;
a = pair<int, int>(num, Queens[num]);
//判断第num个皇后与前面皇后的兼容问题
bool sgn = true;
for (j = 0; j < num; j++)
{
b = pair<int, int>(j, Queens[j]);
if (!isOk(a, b)){ sgn = false; break; }
}
if (sgn == true)
{
vectorRoad;
for (int j = 1; j <= N; j++)
Road.push_back(Queens[j-1]);
res.push_back(Road);
}
}
}
return ;
}
};