蓝桥杯 - 受伤的皇后

解题思路:

递归 + 回溯(n皇后问题的变种)

在 N 皇后问题的解决方案中,我们是从棋盘的顶部向底部逐行放置皇后的,这意味着在任何给定时间,所有未来的行(即当前行之下的所有行)都还没有被探查或放置任何皇后。因此,检查下方行是没有意义的,因为它们总是空的。所以只需要检查左上45°和右上45°。

java 复制代码
import java.util.Scanner;

public class Main {
    static int count = 0;

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int n = scan.nextInt();
        int[][] arr = new int[n][n];
        dfs(arr, 0);
        System.out.println(count);
    }

    public static void dfs(int[][] arr, int row) {
        if (row == arr.length) {
            count++;
            return;
        }
        // 遍历列,因为n行n列,所以arr.length和arr[0].length是一样的
        for (int j = 0; j < arr.length; j++) {
            if (checkValid(arr, row, j)) {
                arr[row][j] = 1;
                dfs(arr, row + 1);
                // 回溯
                arr[row][j] = 0;
            }
        }
    }

    public static boolean checkValid(int[][] arr, int row, int col) {
        // 检查列,因为n行n列,所以row既是行的长度又是列的长度
        for (int i = 0; i < row; i++) {
            if (arr[i][col] == 1) {
                return false;
            }
        }
        // 检查左上45°
        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
            if (arr[i][j] == 1 && Math.abs(row - i) < 3) {
                return false;
            }
        }
        // 检查右上45°
        for (int i = row - 1, j = col + 1; i >= 0 && j < arr.length; i--, j++) {
            if (arr[i][j] == 1 && Math.abs(row - i) < 3) {
                return false;
            }
        }
        return true;
    }
}
相关推荐
非ban必选19 分钟前
spring-ai-alibaba第七章阿里dashscope集成RedisChatMemory实现对话记忆
java·后端·spring
A旧城以西24 分钟前
数据结构(JAVA)单向,双向链表
java·开发语言·数据结构·学习·链表·intellij-idea·idea
杉之41 分钟前
选择排序笔记
java·算法·排序算法
Naive_71 小时前
蓝桥杯准备(前缀和差分)
java·职场和发展·蓝桥杯
烂蜻蜓1 小时前
C 语言中的递归:概念、应用与实例解析
c语言·数据结构·算法
OYangxf1 小时前
图论----拓扑排序
算法·图论
我要昵称干什么1 小时前
基于S函数的simulink仿真
人工智能·算法
头顶秃成一缕光1 小时前
若依——基于AI+若依框架的实战项目(实战篇(下))
java·前端·vue.js·elementui·aigc
AndrewHZ2 小时前
【图像处理基石】什么是tone mapping?
图像处理·人工智能·算法·计算机视觉·hdr
念九_ysl2 小时前
基数排序算法解析与TypeScript实现
前端·算法·typescript·排序算法