[Java] 蓝桥杯ALGO-125 算法训练王、后传说

问题描述
地球人都知道，在国际象棋中，后如同太阳，光芒四射，威风八面，它能控制横、坚、斜线位置。
看过清宫戏的中国人都知道，后宫乃步步惊心的险恶之地。各皇后都有自己的势力范围，但也总能找到相安无事的办法。
所有中国人都知道，皇权神圣，伴君如伴虎，触龙颜者死……
现在有一个n*n的皇宫，国王占据他所在位置及周围的共9个格子，这些格子皇后不能使用（如果国王在王宫的边上，占用的格子可能不到9个）。当然，皇后也不会攻击国王。
现在知道了国王的位置（x,y）（国王位于第x行第y列，x,y的起始行和列为1），请问，有多少种方案放置n个皇后，使她们不能互相攻击。
输入格式
一行，三个整数，皇宫的规模及表示国王的位置
输出格式
一个整数，表示放置n个皇后的方案数
样例输入
8 2 2
样例输出
10
数据规模和约定
n<=12

package algo125;

import java.util.Scanner;

public class Main {
    
    private static int[] pos;
    private static int x;
    private static int y;
    private static int cnt;
    
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        pos = new int[n];
        x = in.nextInt() - 1;
        y = in.nextInt() - 1;
        in.close();
        
        dfs(0, n);
        System.out.println(cnt);
    }
    
    private static boolean isSafe(int row) {
        if (Math.abs(row - x) <= 1 && Math.abs(pos[row] - y) <= 1) {
            return false;
        }
        
        for (int i = 0; i < row; i++) {
            if (pos[row] == pos[i] || row - i == Math.abs(pos[row] - pos[i])) {
                return false;
            }
        }
        return true;
    }
    
    private static void dfs(int row, int n) {
        if (row >= n) {
            cnt++;
            return;
        }
        
        for (pos[row] = 0; pos[row] < n; pos[row]++) {
            if (isSafe(row)) {
                dfs(row + 1, n);
            }
        }
    }

}

package algo125;

import java.util.Scanner;

public class Main {

private static int[] pos;

private static int x;

private static int y;

private static int cnt;

public static void main(String[] args) {

Scanner in = new Scanner(System.in);

int n = in.nextInt();

pos = new int[n];

x = in.nextInt() - 1;

y = in.nextInt() - 1;

in.close();

dfs(0, n);

System.out.println(cnt);

}

private static boolean isSafe(int row) {

if (Math.abs(row - x) <= 1 && Math.abs(pos[row] - y) <= 1) {

return false;

}

for (int i = 0; i < row; i++) {

if (pos[row] == pos[i] || row - i == Math.abs(pos[row] - pos[i])) {

return false;

}

return true;

}

private static void dfs(int row, int n) {

if (row >= n) {

cnt++;

return;

}

for (pos[row] = 0; pos[row] < n; pos[row]++) {

if (isSafe(row)) {

dfs(row + 1, n);

}

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版