原题链接:蓝桥杯基础练习VIP-2n皇后问题
package practise;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Scanner;
public class Main {
static Scanner sc = new Scanner(new BufferedReader(new InputStreamReader(System.in)));
public static void main(String[] args) {
n = sc.nextInt();
rec_b = new int[n];
rec_w = new int[n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
map[i][j] = sc.nextInt();
}
}
dfs_w(0);
System.out.println(cnt);
}
static int n;
static int cnt = 0;//结果
static int[] rec_b;//黑皇后
static int[] rec_w;//白皇后
static int[][] map = new int[10][10];
//白皇后dfs
private static void dfs_w(int row) {
// TODO Auto-generated method stub
if(row == n) {
dfs_b(0);
}
for (int col = 0; col < n; col++) {
if(map[row][col]==1 && check_w(row,col)) {
map[row][col] = 0;
rec_w[row] = col;
dfs_w(row+1);
map[row][col] = 1;
rec_w[row] = 0;
}
}
}
//判断白皇后位置
private static boolean check_w(int row, int col) {
for (int i = 0; i < row; i++) {
if(rec_w[i]==col||Math.abs(rec_w[i]-col)==Math.abs(i-row)) {
return false;
}
}
return true;
}
//黑皇后dfs
private static void dfs_b(int row) {
// TODO Auto-generated method stub
if(row == n) {
cnt++;
return;
}
for (int col = 0; col < n; col++) {
if(map[row][col]==1 && check_b(row,col)) {
map[row][col] = 0;
rec_b[row] = col;
dfs_b(row+1);
//回溯
map[row][col] = 1;
rec_b[row] = 0;
}
}
}
//判断黑皇后位置
private static boolean check_b(int row, int col) {
for (int i = 0; i < row; i++) {
if(rec_b[i]==col||Math.abs(col-rec_b[i])==Math.abs(i-row)) {
return false;
}
}
return true;
}
} 0.0分
0 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复