蓝桥杯 算法训练 N皇后问题（Java）

问题描述：

在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。你的任务是，对于给定的N，求出有多少种合法的放置方法。

输入格式：

输入中有一个正整数N≤10，表示棋盘和皇后的数量

输出格式：

为一个正整数，表示对应输入行的皇后的不同放置数量。

样例输入：

5

样例输出：

10

数据规模和约定：

N≤10

思路：

1. 题目中数模的大小是<=10，我们可以提前准备好数组，由于下面我是从1开始赋值的，因此我数组的大小设置初始值为11。

2. 由题可知，皇后的放置位置不允许放在同一行同一列以及斜对角线，而且每个皇后开始的位置不同，那么得到的结果就不相同。所以每轮完一次我们就必须要回溯一次也就是把最先前已经放置好的那个位置进行清零，换下一个位置进行放置。

3. 当超出边界的时候，并且皇后已经全部放置完成，计数器才自加1.

完整代码：

import java.util.*;

public class Main {

    static int n; //输入皇后的数量
    static int count; //计算全过程满足条件的放置方式
    static int[] a = new int[11]; //设置数组，代表皇后在的行数
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        dfs(1);//从第一行开始放置皇后
        System.out.println(count);
    }

    static void dfs(int row){

        if(row == n+1){ //判断皇后是否已经全部放置完，到达边界
            count++;  //只有放好了这一行的皇后才能放置下一行的
            return;
        }

        for (int i = 1; i <= n; i++){ //i 表示这一行的列
            if (check(row,i)){ //判断这所有行的第i列在本次情况中是否已经放置好了皇后
                a[row]=i;  // 将皇后放入符合条件的列数中
                dfs(row+1); //然后放置下一个皇后
                a[row]=0;  //当n中情况的全部皇后放置完成后，把之前的所有皇后放置的位置清零，重新重第一行的下一列的位置开始。
            }
        }
    }

    static boolean check(int x, int y){
        for (int i = 1; i <= x; i++){ // 表示从第一行开始，到目前想要放置皇后的行数
            if (a[i] == y) return false; 
            if (i+a[i] == x+y) return false; 
            if (i-a[i] == x-y) return false; 
        }
        return true;
    }


    }

测试结果：