【回溯法】八皇后问题
问题描述
在一个8*8的棋盘中,有八个皇后的棋子。这些棋子所放的位置的同一行,同一列和同一个斜线上不能出现另一个皇后,问有多少种摆放的方式。
示例
输出:
92
解题思路
采用回溯法,分别有递归和迭代两种方式:
1、递归:
当8个皇后还没有都分配时,对其进行分配,具体如下:
从第一列开始遍历,对当前皇后进行分配,如果当前列没有其他的皇后,那么就分配成功,然后进入递归进行下一行皇后的分配。如果分配失败,那就进入下一次遍历,重复上述过程。
当8个皇后都分配完毕后,进行一次次数统计。
2、迭代:
对当前的皇后k进行列的选择,选择当前能选的最小的列,然后当8个皇后都选完后,进行次数+1,如果出现有皇后选择的行超过7就进行回溯。
代码
public class EightQueen {
static int x[]; //存放第k个皇后所在列号
static int sum;
public static void main(String[] args) {
x = new int[8];
for(int i=0;i<8;++i){
x[i] = -1;
}
queen(8);
//dfs(0);
System.out.println(sum);
}
//检查当前皇后是否与前面的冲突(同列或同一斜线)
static int check(int k){
for(int i=0;i<k;++i)
if(x[i] == x[k] || Math.abs(x[i]-x[k]) == Math.abs(i-k)) //在同一列或同一斜线
return 1;
return 0;
}
//迭代
static void queen(int n){
int k = 0;
sum = 0;
while(k >= 0){
x[k]++; //摆放第k个皇后
while(x[k]<n && check(k)==1)
x[k]++;
if(x[k]< n && k==n-1){
// 8皇后全放进去了
for(int i=0;i<n;++i)
System.out.print((x[i]+1)+" ");
System.out.println();
sum++;
}
if(x[k]<n && k<n-1) //下一个皇后
++k;
else{ //回溯
x[k]=-1;
--k;
}
}
}
//递归
static void dfs(int step) {
if(step == 8){
for(int i=0;i<8;++i)
System.out.print((x[i]+1)+" ");
System.out.println();
sum++;
}
else{
for(int i = 0;i < 8;++i){
x[step] = i;
if(1 != check(step)){
dfs(step+1);
}
}
}
}
}