问题:任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法?
算法思路分析
1) 第一个皇后先放第一行第一列
2) 第二个皇后放在第二行第一列、然后判断是否OK,如果不行,继续放在第二列、第三列,一次把所有列都放完,找到一个合适
3)继续第三个皇后,还是第一列、第二列…直到第八个皇后也能放在一个正确的位置上,得出一个正确解
4)当得到一个正确解时,在栈回退到上一个栈时,就会开始回溯,将第一个皇后,放到第一列的所有正确解,全部得到
5)然后回头继续第一个皇后放在第二列,继续循环执行1,2,3,4的步骤
注:理论上应该创建一个二维数组表示棋盘,但是我们可以通过算法,用一个一维数组来解决问题,arr[8]={0,4,7,5,2,6,1,3}
代码实现
package recursion;
public class queue8 {
//定义一个max表示共有多少个皇后
int max = 8;
//定义数组array,保存皇后放置位置的结果,比如arr={0,4,7,5,2,6,1,3}
int[] array = new int[max];
static int count =0;
static int judgeCount =0;
public static void main(String[] args) {
queue8 queue8 = new queue8();
queue8.check(0);
System.out.println("解法共有:"+count);
System.out.println("判断次数:"+judgeCount);
}
//放置第n个皇后
private void check(int n){
if(n==max){//n=8
print();
return;
}
//依次放入皇后,并判断是否冲突
for (int i = 0; i <max; i++) {
//先把当前皇后放到该行的第一列
array[n]=i;
//判断当前放置第n个皇后到i列时,是否冲突
if (judge(n)) {//不冲突
//接着放n+1个皇后,开始递归
check(n+1);
}
//如果冲突,就继续执行array[n]=i;将第n个皇后放置到本行后一个位置
}
}
//查看当我们放置第n个皇后时,就去检测该皇后是否和前面已经摆放的皇后冲突
/**
* @param n 表示第n个皇后
* @return
*/
private boolean judge(int n) {
judgeCount++;
for (int i = 0; i < n; i++) {
//array[i] == array[n]判断是否在同一列
//Math.abs(n-i)==Math.abs(array[n]-array[i])判断是否在同一斜线
if (array[i] == array[n] ||Math.abs(n-i)==Math.abs(array[n]-array[i])){
return false;
}
}
return true;
}
//写一个方法,可以将皇后摆放的位置输出
private void print() {
for (int i = 0; i < array.length; i++) {
System.out.print(array[i] + " ");
}
count++;
System.out.println();
}
}
算法分析
共判断次数:15720,说明回溯算法的效率并不高,可以通过贪心算法提高效率。