八皇后问题,一个古老而著名的问题,是回溯算法的典型案例。该问题由国际西洋棋棋手马克斯·贝瑟尔于 1848 年提出:在 8×8 格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。高斯认为有 76 种方案。1854 年在柏林的象棋杂志上不同的作者发表了 40 种不同的解,后来有人用图论的方法解出 92 种结果。计算机发明后,有多种计算机语言可以编程解决此问题。
输入: 4
输出: [
[".Q..", // 解法 1
"...Q",
"Q...",
"..Q."],
["..Q.", // 解法 2
"Q...",
"...Q",
".Q.."]
]
针对N皇后问题,若要求解出所有可能的解,这必须列举出N个皇后位置所有的可能组合。一种求解方式就是把N个皇后在棋盘中的所有位置的组合都列举出来,然后依次判断每一个组合会不会出现攻击。这种方法的时间复杂度 O(N^N),显然复杂度是相当高的。
为了能够简化时间复杂度,我们在按照一行一行的当时去放置N个皇后,每行都会放置一个皇后,并且该行放置皇后的位置最大为N;一旦本行的皇后确定了位置后,那么 她所在的行(row),列(col),左斜(hill),右斜(dale) 都不能再放皇后,这样,考虑下一行的皇后的位置时可能的位置就必须避开这些位置。
对于第一行,有N个位置可以放置皇后;
对于第二行,有N - 3个位置可以放置皇后;
因此,我们一行一行的放置皇后的位置,如果发现该行没有可以放置皇后的位置时,则回溯到上一行重新放置上一行的皇后的位置。这就是回溯算法。
实现
观察棋盘,一共有 N 个row, N 个col,2N - 1 个hill,2N - 1 个dale。因此,为了能够记录哪一行、列、斜不能放置皇后,引入数组 记录:
int n;
int[] rows; // 所在行是否有queen
int[] hills; // 左斜是否有queen
int[] dales; // 右斜是否有queen
int[] queens; // 每列queen 放置的位置
判断 该位置 是否能放置皇后
private boolean isAttack(int row, int col){
int res = rows[col] + hills[row - col + 2 * n] + dales[row + col];
return res == 0;
}
放置皇后, 在位置(row,col)放置皇后后,需更新 rows, hill, dales数组
private void placeQueen(int row, int col){
queens[row] = col;
rows[col] = 1;
hills[row - col + 2 * n] = 1;
dales[col + row] = 1;
}
移走皇后,在位置(row,col)移走皇后后,需更新 rows, hill, dales数组
private void removeQueen(int row, int col){
queens[row] = 0;
rows[col] = 0;
hills[row - col + 2 * n] = 0;
dales[col + row] = 0;
}
所有位置都放置好后,生成所需要的字符串列表
private void addSolution(){
List<String> solution = new ArrayList<>();
for(int i = 0; i < n; i ++){
int col = queens[i];
StringBuilder sb = new StringBuilder();
for(int j = 0; j < col; j ++)
sb.append(".");
sb.append("Q");
for(int j = col + 1; j < n; j ++)
sb.append(".");
solution.add(sb.toString());
}
output.add(solution);
}
核心回溯算法
private void backtrace(int row){
for(int col = 0; col < n; col ++){
if(isAttack(row, col)){
placeQueen(row, col);
if(row == n - 1)
addSolution();
else
backtrace(row + 1);
removeQueen(row, col);
}
}
}
class Solution {
int[] rows; // 所在行是否有queen
int[] hills; // 左斜是否有queen
int[] dales; // 右斜是否有queen
int n;
int[] queens; // 每列queen 放置的位置
List<List<String>> output;
public List<List<String>> solveNQueens(int n){
this.n = n;
rows = new int[n];
hills = new int[4*n - 1];
dales = new int[2*n - 1];
queens = new int[n];
output = new ArrayList<>();
backtrace(0);
return output;
}
private boolean isAttack(int row, int col){
int res = rows[col] + hills[row - col + 2 * n] + dales[row + col];
return res == 0;
}
private void placeQueen(int row, int col){
queens[row] = col;
rows[col] = 1;
hills[row - col + 2 * n] = 1;
dales[col + row] = 1;
}
private void removeQueen(int row, int col){
queens[row] = 0;
rows[col] = 0;
hills[row - col + 2 * n] = 0;
dales[col + row] = 0;
}
private void addSolution(){
List<String> solution = new ArrayList<>();
for(int i = 0; i < n; i ++){
int col = queens[i];
StringBuilder sb = new StringBuilder();
for(int j = 0; j < col; j ++)
sb.append(".");
sb.append("Q");
for(int j = col + 1; j < n; j ++)
sb.append(".");
solution.add(sb.toString());
}
output.add(solution);
}
private void backtrace(int row){
for(int col = 0; col < n; col ++){
if(isAttack(row, col)){
placeQueen(row, col);
if(row == n - 1)
addSolution();
else
backtrace(row + 1);
removeQueen(row, col);
}
}
}
}