原题: N皇后
n 皇后问题 研究的是如何将 n
个皇后放置在 n×n
的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n
,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q'
和 '.'
分别代表了皇后和空位。
示例 1:
输入:n = 4
输出:[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释:如上图所示,4 皇后问题存在两个不同的解法。
示例 2:
输入:n = 1
输出:[["Q"]]
提示:
1 <= n <= 9
- 皇后彼此不能相互攻击,也就是说:任何两个皇后都不能处于同一条横行、纵行或斜线上。
思路:
什么是n皇后?
在n×n格的棋盘上放置彼此不受攻击的n个皇后。按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。n皇后问题等价于再n×n的棋盘上放置n个皇后,任何2个皇后不妨在同一行或同一列或同一斜线上。剪枝 + 回溯 解决
思路:
1、查找第一行,从左往右查找,发现合适的位置,放下第一个皇后
2、查找第二行,从左往右查找,发现合适的位置,放下第二个皇后
3、查找第三行,从左往右查找,发现合适的位置,放下第三个皇后
。
。
。
n、查找第n行,从左往右查找,发现合适的位置,放下第n个皇后如何判断合适的位置?
假设要在m行k列放置一个皇后,只需要查看第一行到m-1行就行。如果k列已经存在皇后,那么m行k列这个位置就不适合了,如果这个位置的对角线上有其他皇后了,同样这个位置也不适合了。
方法一:递归 + 回溯+ 剪枝
public List<List<String>> solveNQueens(int n) {
queen(n);
return res;
}
List<List<String>> res = new ArrayList<>();
//数组的索引为行号,值为列号
int[] cols;
//n皇后的摆法总数
int ways;
public void queen(int n) {
if (n < 1) {
return;
}
cols = new int[n];
place(0);
}
//在第row行放置皇后
private void place(int row) {
//如果全部放完,则记录下一种摆法
if (row == cols.length) {
ways++;
res.add(generateAns());
return;
}
//遍历当前行的每一列
for (int i = 0; i < cols.length; i++) {
//如果第row行第i列可以放皇后
if (isValid(row, i)) {
cols[row] = i;
//递归在row+1行放置皇后
place(row + 1);
}
}
}
//判断在row行col列是否可以放置皇后
private boolean isValid(int row, int col) {
for (int i = 0; i < row; i++) {
//如果col列已经被row前面的行放置过皇后了,说明这个位置不能放置了
if (cols[i] == col) {
return false;
}
//row行col列这个位置的对角线存在别的皇后了,说明这个位置不能放置了
//利用斜率 两个点的行的差值 等于 列的差值, 对角线斜率为 |k| = 1
if (row - i == Math.abs(col - cols[i])) {
return false;
}
}
return true;
}
//记录每一种摆法
public List<String> generateAns() {
List<String> res = new ArrayList<>();
for (int row = 0; row < cols.length; row++) {
StringBuilder sb = new StringBuilder();
for (int col = 0; col < cols.length; col++) {
if (cols[row] == col) {
sb.append("Q");
} else {
sb.append(".");
}
}
res.add(sb.toString());
}
return res;
}
方法二:位运算(后续更新…)