蓝桥杯 基础训练 2n皇后问题 Java

基础练习 2n皇后问题

资源限制

时间限制：1.0s 内存限制：512.0MB

问题描述

给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。

输入格式

输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。

输出格式

输出一个整数，表示总共有多少种放法。

样例输入

4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样例输出

2

样例输入

4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样例输出

0

参考代码

import java.util.Scanner;

public class Main {
	static int n;
	static int[][] a;
	static int ans = 0;
	static boolean[] b = new boolean[20]; // 标记黑皇后占领的列
	static boolean[] c = new boolean[20]; // 标记黑皇后占领的对角线(左下到右上)
	static boolean[] d = new boolean[20]; // 标记黑皇后占领的对角线(左上到右下)
	static boolean[] b2 = new boolean[20]; // 标记白皇后占领的列
	static boolean[] c2 = new boolean[20]; // 标记白皇后占领的对角线(左下到右上)
	static boolean[] d2 = new boolean[20]; // 标记白皇后占领的对角线(左上到右下)

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		a = new int[n + 1][n + 1];
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= n; j++) {
				a[i][j] = sc.nextInt();
			}
		}
		dfs(1);
		System.out.println(ans);
	}

	private static void dfs(int x) {
		if (x == n + 1) {
			ans++;
			return;
		}
		for (int i = 1; i <= n; i++) {
			// 判断该位置是否可以放置黑皇后
			if (a[x][i] == 1 && !b[i] && !c[x + i] && !d[x - i + n]) {
				a[x][i] = 0; //此位置被皇后占领
				b[i] = true; // 此列被黑皇后占领
				c[x + i] = true; // 对角线被黑皇后占领
				d[x - i + n] = true; // 对角线被黑皇后占领 ("+n"防止数组下标为负数)		
				for (int j = 1; j <= n; j++) {
					// 判断该位置是否可以放置白皇后
					if (a[x][j] == 1 && !b2[j] && !c2[x + j] && !d2[x - j + n]) {
						a[x][j] = 0;  //此位置被皇后占领
						b2[j] = true; // 此列被白皇后占领
						c2[x + j] = true; // 对角线被白皇后占领
						d2[x - j + n] = true; // 对角线被白皇后占领 ("+n"防止数组下标为负数)
						dfs(x + 1);
						// 回溯
						a[x][j] = 1;
						b2[j] = false;
						c2[x + j] = false;
						d2[x - j + n] = false;
					}
				}
				// 回溯
				a[x][i] = 1;
				b[i] = false;
				c[x + i] = false;
				d[x - i + n] = false;
			}
		}
	}
}