当解决N皇后问题时,我们的目标是将N个皇后放置在一个NxN的棋盘上,使得它们互相之间无法互相攻击。皇后可以在同一行、同一列和同一对角线上攻击其他皇后。
下面是一个用Java实现的N皇后问题的算法,我将逐行解析代码:
import java.util.ArrayList;
import java.util.List;
public class NQueens {
public List<List<String>> solveNQueens(int n) {
List<List<String>> result = new ArrayList<>();
int[] queens = new int[n];
solveNQueensHelper(result, queens, 0, n);
return result;
}
private void solveNQueensHelper(List<List<String>> result, int[] queens, int row, int n) {
if (row == n) {
// 找到一个解,将结果添加到最终结果列表中
result.add(generateBoard(queens));
} else {
for (int col = 0; col < n; col++) {
if (isValid(queens, row, col)) {
queens[row] = col; // 在当前行的col列放置皇后
solveNQueensHelper(result, queens, row + 1, n); // 递归调用下一行
}
}
}
}
private boolean isValid(int[] queens, int row, int col) {
for (int i = 0; i < row; i++) {
if (queens[i] == col || Math.abs(queens[i] - col) == row - i) {
// 在同一列或同一对角线上已经有皇后
return false;
}
}
return true;
}
private List<String> generateBoard(int[] queens) {
List<String> board = new ArrayList<>();
for (int i = 0; i < queens.length; i++) {
char[] row = new char[queens.length];
Arrays.fill(row, '.');
row[queens[i]] = 'Q';
board.add(new String(row));
}
return board;
}
}
现在我们逐个解析代码:
- 首先,我们引入了必要的类
ArrayList
和List
,它们用于存储和操作结果。 - 然后,我们定义了一个名为
NQueens
的类。 - 在
NQueens
类中,我们有一个solveNQueens
方法,它接受一个整数n
作为参数,并返回一个列表result
,其中包含所有的解。 - 在
solveNQueens
方法内部,我们创建一个空的列表result
用于存储最终结果。 - 接下来,我们创建一个长度为
n
的整数数组queens
,用于存储皇后在每一行的列索引。 - 我们调用
solveNQueensHelper
方法来寻找解,传递result
、queens
、0
和n
作为参数。 solveNQueensHelper
方法是一个递归函数,
它使用回溯法来解决N皇后问题。回溯法是一种通过尝试所有可能的选择来找到解的方法。
-
首先,我们检查如果
row
等于n
,表示我们找到了一个解,将当前皇后的位置转换为棋盘形式,并将其添加到最终结果列表result
中。 -
否则,我们使用一个循环来尝试在当前行的每一列放置皇后。
-
对于每一列,我们通过调用
isValid
方法来检查当前位置是否是一个有效的选择。 -
isValid
方法用于检查当前位置是否与之前的皇后位置冲突,即是否在同一列或同一对角线上。 -
如果位置有效,我们将当前位置的列索引存储在
queens
数组的row
位置上,并递归调用solveNQueensHelper
方法处理下一行。 -
递归调用会逐行进行,直到找到所有的解。
-
最后,我们返回最终结果列表
result
。 -
在
solveNQueensHelper
方法之外,我们还定义了一个辅助方法generateBoard
,用于将皇后的位置转换为棋盘形式的字符串列表。 -
generateBoard
方法遍历queens
数组中的每个位置,将每一行转换为一个字符数组,并使用'.'
填充所有位置。 -
在皇后的位置上,我们将字符数组中的对应位置修改为
'Q'
,表示皇后的位置。 -
最后,我们将每一行的字符数组转换为字符串,并将其添加到结果列表
board
中。 -
最后,我们返回结果列表
board
。
这就是N皇后问题的Java代码实现及其详细解析。这个算法使用回溯法来逐行尝试放置皇后,并使用有效性检查来确保皇后之间没有冲突。通过递归调用和回溯,算法能够找到所有的解,并将它们转换为棋盘形式的字符串列表。