Java学习——数据结构——递归实现八皇后问题

学习尚硅谷韩顺平老师的Java数据结构笔记，详情请移步网站

1. 八皇后问题介绍
   八皇后问题，是一个古老而著名的问题，是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯·贝瑟尔于1848 年提出：在 8×8 格的国际象棋上摆放八个皇后，使其不能互相攻击，即：任意两个皇后都不能处于同一行、同一列或同一斜线上，问有多少种摆法。
2. 八皇后问题算法思路分析
   （1) 第一个皇后先放第一行第一列
   （2) 第二个皇后放在第二行第一列、然后判断是否 OK， 如果不 OK，继续放在第二列、第三列、依次把所有列都放完，找到一个合适
   （3) 继续第三个皇后，还是第一列、第二列……直到第 8 个皇后也能放在一个不冲突的位置，算是找到了一个正确解
   （4) 当得到一个正确解时，在栈回退到上一个栈时，就会开始回溯，即将第一个皇后，放到第一列的所有正确解， 全部得到.
   （5) 然后回头继续第一个皇后放第二列，后面继续循环执行 1,2,3,4 的步骤
   说明：
   理论上应该创建一个二维数组来表示棋盘，但是实际上可以通过算法，用一个一维数组即可解决问题. arr[8] =
   {0 , 4, 7, 5, 2, 6, 1, 3} //对应 arr 下标 表示第几行，即第几个皇后，arr[i] = val , val 表示第 i+1 个皇后，放在第 i+1
   行的第 val+1 列

package com.Recursion;

/**
 * 递归实现八皇后问题
 */
public class Queue8 {

    //定义一个max表示共有多少个皇后
    int max = 8;
    //定义数组array, 保存皇后放置位置的结果, 比如arr = {0, 4, 7, 5, 2, 6, 1, 3}
    //对应array下标表示第几行，即第几个皇后，array[i]=value , value表示第i+1个皇后，放在第i+1行的第value+1列
    int[] array = new int[max];
    private static int count = 0;

    public static void main(String[] args) {
        Queue8 queue8 = new Queue8();
        queue8.check(0);
        System.out.println(count);
    }

    /**
     * 放置第n个皇后
     * check是每一次递归时，进入到check中都有for(int i = 0; i < max; i++)，因此会有回溯
     * @param n
     */
    private void check(int n){
        if (n == max){//当n=8, 其实8个皇后就已经放好了
            print();
            return;
        }

        //依次放入皇后，并判断是否冲突
        for (int i = 0; i < max; i++) {
            //先把当前皇后放到当前行的第i列
            array[n] = i;
            //判断当前皇后放到i列时是否冲突
            if (judge(n)){
                //接着放下一个皇后
                check(n+1);
            }
            //如果冲突，就继续执行array[n] = i，即将当前皇后，在本行后移一个位置
        }
    }

    /**
     * 判断当前皇后是否和之前的n-1个皇后冲突
     * @param n 表示第n个皇后
     * @return
     */
    private boolean judge(int n){
        for (int i = 0; i < n; i++) {
            //array[i] == array[n] 表示判断第n个皇后是否和前面的n-1个皇后在同一列
            //Math.abs(n-i) == Math.abs(array[n] - array[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();
    }
}