用递归解决八皇后问题
问题描述
八皇后问题是国际西洋棋棋手马克斯·贝瑟尔于1848年提出:在8X8格的国际象棋上摆放八个皇后,使其不能互相攻击,即:任意两个皇后都不能处于同一行,同一列或者同一斜线上,问由多少种摆法。
思路分析
- 第一个皇后先放在第一行的第一列
- 第二个皇后放在第二行的第一列,然后进行判断。判断什么呢?判断与前面所有的皇后是否会互相攻击。如果会互相攻击则换成第二列、第三列直到把所有的列放完,如果所有的列都放完还没有正确的解,则回溯上一个皇后的摆法。
- 重复第三步的动作,直到8个皇后全部摆完,就得到一个解。
- 当得到一个正确的解之后,在栈中回退到上一个栈,又开始回溯,即将第一个皇后放在第一列的所有解都得到。
- 然后回头继续第一个皇后放第二列,后面继续虚幻执行1,2,3,4的步骤。
代码实现
理论上我们可以使用一个二维数组来表示一个完整棋盘,但是我们也可以通过用一个一维数组来解决这个问题。
即:用数组的下标来表示棋盘的行+1,arr[i]=val,val表示第i+1个皇后,放在第i+1行的第val+1列。
输出解法
private void print(){
count++;
for (int i = 0; i <array.length ; i++) {
System.out.print(array[i]+" ");
}
System.out.println();
}
由于是一个一维数组,所以输出简单。
判断皇后是否会互相攻击
这里的n表示的是第n个皇后, 该方法是查看当我们放置的第n个皇后,就去检测该皇后是否和前面已经摆放的皇后冲突。
private boolean judge(int n){
for (int i = 0; i <n ; i++) {
if (array[i]==array[n]||Math.abs(n-i)==Math.abs(array[n]-array[i])){
return false;
}
}
return true;
}
- array[i]==array[n] 表示判断第n个皇后是否和前面的n-1个皇后在同一列.
- Math.abs(n-i)==Math.abs(array[n]-array[i])表示第n个皇后是否和第i个皇后在同一个斜线上。(就像一个等腰直角三角形,在同一斜线时 ,两个直角边相等。
- 即横着的直角边等于两个皇后的列数相减的绝对值(Math.abs(array[n]-array[i]))、竖着的直角边为两个皇后的行数相减的绝对值Math.abs(n-i)。
放置皇后
这个方法是用来放置皇后的,n就是指放置的第n个皇后。
private void check(int n){
if (n==max){
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=8时,说明全部皇后都已经放置完毕,就直接输出解法。
- 当不等于8时,就进入for循环。( 特别注意:check是每一次递归时,进入到check中都有for (int i = 0; i <max ; i++) ,因此会有回溯)。
- 进入for循环后,先把当前这个皇后n,放到改行的第1列,然后判断是否冲突。
- 不冲突则接着放第n+1个皇后,开始递归。
- 冲突的话,i++之后就继续执行array[n=]i;将该皇后放到后面一列。
完整代码
package com.glk.recursion;
public class Queen {
//d定义一个max表明一共有多少个皇后
int max=8;
// 这里数组的下标表示皇后的行数,下标对应的值表示皇后的列数
// 定义数组array,保存皇后放置位置的结果,比如arr={0,4,7,5,2,6,1,3}
int [] array=new int[max];
static int count=0;
public static void main(String[] args) {
// 测试
Queen queen = new Queen();
queen.check(0);
System.out.printf("一共%d种解法",count);
}
// 编写一个方法,放置第n个皇后
// 特别注意:check是每一次递归时,进入到check中都有for (int i = 0; i <max ; i++) ,因此会有回溯
private void check(int n){
// n=8,8个皇后放好
if (n==max){
print();
return;
}
for (int i = 0; i <max ; i++) {
// 先把当前这个皇后n,放到改行的第1列
array[n]=i;
// 判断当前放置第n个皇后到第i列的时候是否冲突
if (judge(n)){ //不冲突
// 接着放n+1个皇后,即开始递归
check(n+1);
}
// 如果冲突,就继续执行array[n]=i;,即将第n个皇后,放置到本行的后一位置
}
}
// 查看当我们放置的第n个皇后,就去检测该皇后是否和前面已经摆放的皇后冲突
// 表示第几个皇后
private boolean judge(int n){
/*
* 1、array[i]==array[n] 表示判断第n个皇后是否和前面的n-1个皇后在同一列
* 2、Math.abs(n-i)==Math.abs(array[n]-array[i])表示第n个皇后是否和第i个皇后在同一个斜线上(就像一个等腰直角三角形,在同一斜线时
* ,两个直角边相等,即横着的直角边等于两个皇后的列数相减的绝对值(Math.abs(array[n]-array[i]))、竖着的直角边为两个皇后的行数相减的绝对值Math.abs(n-i))
* */
for (int i = 0; i <n ; i++) {
if (array[i]==array[n]||Math.abs(n-i)==Math.abs(array[n]-array[i])){
return false;
}
}
return true;
}
// 写一个方法把皇后摆放的位置输出
private void print(){
count++;
for (int i = 0; i <array.length ; i++) {
System.out.print(array[i]+" ");
}
System.out.println();
}
}
总结
八皇后是一个非常经典用递归解决的问题。本算法主要的难点在于判断什么时候两个皇后会互相攻击,以及什么时候回溯的,在搞明白这两个问题之后,这个就很容易理解了。这个方法也是根据韩顺平老师的数据结构与算法的视频讲解写的,所以还有什么不明白的可以去看一下老师的视频。