刷题蓝桥杯（java）---2N皇后

/*问题描述

　　给定一个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[] bq;// 记录黑皇后的位置
	static int[] wq;// 白皇后
	static int number = 0;
	static int[][] map;

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		bq = new int[n];
		wq = new int[n];
		map = new int[n][n];
		for (int i = 0; i < n; i++) {// 初始化棋盘
			for (int j = 0; j < n; j++)
				map[i][j] = in.nextInt();
		}
		bqueen(0);//0对应0行，t
		System.out.println(number);

	}

	static void wqueen(int t) {
		// 白皇后，先去看黑皇后的，思路是先黑后白
		boolean panduan = false;// 用来判断是否可以放
		for (int i = 0; i < n; i++) {// 找到当前行合适的长度,i表示列
			if (map[t][i] == 1 && bq[t] != i) {// 棋盘上可以放
				for (int k = 0; k < t; k++) {// 遍历之前的皇后并判断当前位置是否可以放，
					// 注意k<t,比较的是当前行之前的皇后
					if (wq[k] == i | Math.abs(k - t) == Math.abs(wq[k] - i)) {// 同列或同对角线，一定不同行
						panduan = true;// 当前位置不能放
						break;
					}

				}
				if (!panduan) {// 能放
					wq[t] = i;
					if (t == n - 1) {// 白皇后到达终点
						number++;
						
					} else
						wqueen(t + 1);// 进入下一步
				} else
					panduan = false;// 复原防止影响下一次的比较
			}
		}
		return;

	}

	static void bqueen(int t) {// 每行的操作
		// 黑皇后
		boolean panduan = false;// 用于标记当前列是否可放
		for (int i = 0; i < n; i++) {// 找到当前行合适的长度,i表示列
			if (map[t][i] == 1) {// 棋盘上可以放
				for (int k = 0; k < t; k++) {// i表示列，遍历之前的皇后并判断当前位置是否可以放，
					// 注意k<t,比较的是当前行之前的皇后
					if (bq[k] == i || Math.abs(k - t) == Math.abs(bq[k] - i)) {// 同列或同对角线，一定不同行
						panduan = true;// 当前位置不能放
						break;
					}

				}
				if (!panduan) {// 能放
					bq[t] = i;
					if (t == n - 1) {// 找到终点加入白皇后
						wqueen(0);// 进入白皇后的放置
					
					} else
						bqueen(t + 1);// 进入下一行
				} else
					panduan = false;// 复原防止影响下一次的比较
			}
		}
		return;// 当前行没有合适的地方可以放置皇后即为函数的正常结束（可不写return）

	}

}