（Java）试题 算法训练 N皇后问题

问题描述

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

输入格式

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

输出格式

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

样例输入

5

样例输出

10

数据规模和约定

N≤10

代码及注释

import java.util.Scanner;

public class Main {
    static int n;
    static char[][] g;//用来存放棋盘
    static boolean[] col,dg,udg;
    static int count = 0;
    static int N = 20;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        g = new char[N][N];
        col = new boolean[N];
        dg = new boolean[N];
        udg = new boolean[N];
        dfs(0);
        System.out.println(count);
    }

    //location 表示遍历到了第几行，即遍历到了第几个皇后
    public static void dfs(int location){
        //当遍历到叶子结点，即得到了一种方案之后
        if (location == n){
            count++;
        }
        for (int i = 0; i < n ; i++) {
            //剪枝:判断是否皇后既不在同一列也不在同一对角线或同一反对角线
            if (!col[i] && !dg[location+i] && !udg[n-location+i]){
                col[i] = dg[location+i] = udg[n-location+i] = true;
                //继续深搜下一行
                dfs(location+1);
                col[i] = dg[location+i] = udg[n-location+i] = false;
            }
        }
    }
}