蓝桥杯----皇后问题

问题描述

　　给定一个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

package com.xjj.lanqiao;

import java.util.Scanner;

//2代表黑皇后放置位置，3代表白皇后放置位置

public class Lq2_27 {
	public static int n;				
	public int[][] a = new int[n][n];		//棋盘
	public int[] book_B =  new int[n];		//标记黑皇后
	public int[] col_B = new int[n];		//黑列
	public int[] book_w =  new int[n];		//标记白皇后
	public int[] col_w = new int[n];		//白列
	public static int count = 0;
	
	//白皇后深度搜索
	public void dfs_W(int index_w){
		if (index_w == n) {
			count++;
			return;
		}
		
		for(int x = 0; x < n; x++){
			if (book_w[x] == 0) {
				boolean flag_w = true;
				if (a[index_w][x] == 1) {
					for(int y = 0; y < index_w; y++){
						if (Math.abs(index_w - y) == Math.abs(x - col_w[y])) {
							flag_w = false;
							break;
						}
					}
					if (flag_w) {
						col_w[index_w] = x;
						a[index_w][x] = 3;
						book_w[x] = 1;
						dfs_W(index_w + 1);
						book_w[x] = 0;
						a[index_w][x] = 1;
					}
				}	
			}
		}
	}
	
	//黑皇后深度搜索
	public void dfs_B(int index_B){
		//黑皇后排列完成，进行白皇后搜索
		if (index_B == n ) {
			dfs_W(0);
		}
		
		//对每个皇后安排位置
		for(int x = 0; x < n; x++){
			//如未被访问
			if (book_B[x] == 0) {
				//回溯法
				boolean flag = true;
				if (a[index_B][x] == 1) {
					for(int y = 0; y < index_B; y++){
						//在对角线
						if (Math.abs(index_B - y) == Math.abs(x - col_B[y])) {
							flag = false;
							break;
						}
					}
					if (flag) {
						col_B[index_B] = x;		//列坐标
						a[index_B][x] = 2;		//将该点标记为黑皇后
						book_B[x] = 1;			//已访问
						dfs_B(index_B + 1);
						book_B[x] = 0;
						a[index_B][x] = 1;
					}
				}
			}
		}
	}

	public static void main(String[] args) {
		System.out.println();
		Scanner scanner = new Scanner(System.in);
		n = scanner.nextInt();
		Lq2_27 lq = new Lq2_27();
		for(int i = 0; i < n; i++)
			for(int j = 0; j < n; j++)
				lq.a[i][j] = scanner.nextInt();
			
		lq.dfs_B(0);
		System.out.println(count);

	}

}