数据结构和算法(38)之八皇后问题

学习数据结构和算法的日常Demo

八皇后问题介绍

八皇后问题，是一个古老而著名的问题，是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯.贝瑟尔于1848年提出:在8X8格的国际象棋上摆放八个皇后，使其不能互相攻击，即:任意两个皇后都不能处于同一行、同一列或同一斜线上，问共有多少种摆法。

思路分析

1. 第一个皇后先放第一行第-列
2. 第二个皇后放在第二行第一 列、然后判断是否OK， 如果不OK，继续放在第二列、第三列、依次把所有列都放完，找到一个合适位置
3. 继续第三个皇后， 还是第一列、第二列…直到第8个皇后也能放在一个不冲突的位置，即找到了一个正确解
4. 当得到一个正确解时，在栈回退到上一个栈时，就会开始回溯，即将第一个皇后，放到第一列的所有正确解，全部得到.
5. 然后回头继续第一个皇后放第二列，后而继续循环执行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();
    }
}

部分结果显示

GitHub:数据结构和算法源代码