n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
上图为 8 皇后问题的一种解法。
给定一个整数 n,返回所有不同的 n 皇后问题的解决方案。每一种解法包含一个明确的 n 皇后问题的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
示例
输入: 4
输出: [
[".Q..", // 解法 1
"...Q",
"Q...",
"..Q."],
["..Q.", // 解法 2
"Q...",
"...Q",
".Q.."]
]
解释: 4 皇后问题存在两个不同的解法。
解法
这个题和37.解数独有异曲同工之处,解数独是行列宫中每个数字只能有一个,N皇后是行列主对角、副对角线中只能有一个皇后。
我们按照解数独的思路去求解这道题
- 生成空棋盘
- 用bool数组表示列、主对角线、副对角线上是否有皇后,将其初始化为false
- 尝试填充棋盘,若列、主对角线、副对角线上有皇后,我们就不填充
- 若填充失败,那么我们需要回溯,将原来尝试填充的地方改回来
- 若填充成功,将棋盘添加至列表中,将列表添加至准备好的 res 中,之后回溯
- 重复3.4.5,直至遍历完第一行
特别注意:
-
如何将同一对角线上的元素表示在 bool[ ] 中?为了在填充皇后后将对角线上的棋盘格子设为true(有皇后,不能填充),这需要我们找到对角线上元素的特性。
如上图所示,我们可以用 paradiagUsed[row + column] 表示副对角线上是否有皇后 用diag[row - column + n -1] 表示主对角线上是否有皇后(+n-1 避免超出索引)。 -
以行为递归单位:当前行填充完成后,下次递归直接对下一行进行填充,因为每行中最多只能有一个皇后,因此不用单独用 bool[ ] 表示行中皇后的情况,只需将下次递归的参数设为row+1即可。
C#代码如下
public IList<IList<string>> SolveNQueens(int n)
{
//生成棋盘
char[][] map = new char[n][];
for (int i = 0; i < map.Length; i++)
{
map[i] = new char[n];
for (int j = 0; j < map[i].Length; j++)
{
map[i][j] = '.';
}
}
//用bool数组表示列、主对角线、副对角线上是否有皇后
bool[] columnUsed = new bool[n];
bool[] diagUsed = new bool[2 * n - 1];
bool[] paradiagUsed = new bool[2 * n - 1];
//回溯法求解
IList<IList<string>> res = new List<IList<string>>();
BackTrack3(res, map, columnUsed, diagUsed, paradiagUsed, 0, n);
return res;
}
public void BackTrack3(IList<IList<string>> res, char[][] map, bool[] columnUsed, bool[] diagUsed,
bool[] paradiagUsed, int row, int n)//回溯法
{
//边界校检
if (row == n)
{
IList<string> list=new List<string>();
for (int i = 0; i < n; i++)
{
list.Add(new string(map[i]));//这里不能用ToString()
}
res.Add(list);
return;
}
//放置皇后
for (int i = 0; i < n; i++)
{
if (!columnUsed[i] && !diagUsed[row-i+n-1] && !paradiagUsed[row+i])
{
columnUsed[i] = true;
diagUsed[row - i + n - 1] = true;
paradiagUsed[row + i] = true;
map[row][i] = 'Q';
BackTrack3(res, map, columnUsed, diagUsed, paradiagUsed, row + 1, n);
//回溯
map[row][i] = '.';
columnUsed[i] = false;
diagUsed[row - i + n - 1] = false;
paradiagUsed[row + i] = false;
}
}
}
对角线图片源自:https://leetcode-cn.com/problems/n-queens/solution/gen-ju-di-46-ti-quan-pai-lie-de-hui-su-suan-fa-si-/
作者:liweiwei1419