学习数据结构和算法的日常Demo
八皇后问题介绍
八皇后问题,是一个古老而著名的问题,是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯.贝瑟尔于1848年提出:在8X8格的国际象棋上摆放八个皇后,使其不能互相攻击,即:任意两个皇后都不能处于同一行、同一列或同一斜线上,问共有多少种摆法。
思路分析
- 第一个皇后先放第一行第-列
- 第二个皇后放在第二行第一 列、然后判断是否OK, 如果不OK,继续放在第二列、第三列、依次把所有列都放完,找到一个合适位置
- 继续第三个皇后, 还是第一列、第二列…直到第8个皇后也能放在一个不冲突的位置,即找到了一个正确解
- 当得到一个正确解时,在栈回退到上一个栈时,就会开始回溯,即将第一个皇后,放到第一列的所有正确解,全部得到.
- 然后回头继续第一个皇后放第二列,后而继续循环执行1.2.3.4的步骤
注:理论上应该创建一个二维数组来表示棋盘,但是实际上可以通过算法用一个一维数组即可解决问题(因为每一行只允许存在一个皇后),例如int arr[]={0,4,7,5,2,6, 1,3} ,arr[i]=val则表示第i+1个皇后,放在第i+1行的第val+1列
代码实现
public class Queen8 {
// 定义一个max数组,表示共有多少皇后
static int max = 8;
// 定义数组arr,保存皇后放置的位置,arr = {0(第一行第一列),5(第二行第六列)}.一行一个皇后,所以存列就行
static int[] arr = new int[max];
// 定义解法计数器
static int count = 0;
public static void main(String args[]) {
check(0);
System.out.println("共有:"+count);
}
// 放置第n个皇后
// 注意:每一次进入check方法,都有for循环,因此产生回溯
private static void check(int n){
if (n == max){
print();
count++; // 计一个解法
return;
}
for (int i = 0; i < max; i++) {
// 先把当前皇后放到该行的第1列
arr[n] = i;
// 判断放到i列时是否冲突
if (judge(n)){ // 不冲突
check(n+1);
}
// 如果冲突,就回到arr[n] = i+1;
}
}
// 当我们放置第n个皇后时,就去检测该皇后是否和已经摆放的皇后冲突(回溯)
private static boolean judge(int n) {
for (int i = 0; i < n; i++) {
// arr[i] == arr[n]判断是否和前面皇后在同一列
// Math.abs(n - i) == Math.abs(arr[n] - arr[i])判断是否和前面皇后在同一斜线
// 二维坐标系中,处于同一斜线上的坐标:(x1,y1),(x2,y2),x1-x2等于y1-y2(绝对值)
if (arr[i] == arr[n] || Math.abs(n - i) == Math.abs(arr[n] - arr[i])) {
return false;
}
}
return true;
}
private static void print() {
for (int i : arr) {
System.out.print(i + " ");
}
System.out.println();
}
}