问题描述
给定一个n*n的棋盘,棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后,使任意的两个黑皇后都不在同一行、同一列或同一条对角线上,任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法?n小于等于8。
输入格式
输入的第一行为一个整数n,表示棋盘的大小。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,如果一个整数为0,表示对应的位置不可以放皇后。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,如果一个整数为0,表示对应的位置不可以放皇后。
输出格式
输出一个整数,表示总共有多少种放法。
样例输入
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
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
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
0
import java.util.*;
//2n皇后问题
public class Main {
public static final int no=0; // 代表不能放棋子
public static final int yes=1; //代表可以放棋子
public static final int hei=2; //代表黑棋子
public static final int bai=3; //代表白棋子
public static int n; // n行n列
public static int[][] qipan=null;//代表棋盘 n*n的二维数组
public static int count=0;//已经放上的当前皇后数
public static int num=0; //摆放的2n皇后的总解
/**
*
* @param qizi 参数棋子
* @return 根据当前的棋子返回相反的棋子 如果当前是黑 就返回白
*/
public static int get(int qizi){
int num=qizi==2? 3: 2;
return num;
}
/**
*
* @param k 行
* @param m 列
* @param qizi 所要检查的棋子
* @return 如果检查存在冲突 返回false 否则返回true
*/
public static boolean jc(int k,int m,int qizi){
if(qipan[k][m]==get(qizi)||qipan[k][m]==no){ //如果当前列已经放了不同棋子或者当前位置是no
return false;
}
for(int i=0;i<n;i++){ //扫描当前列 是否有存在相同棋子
if(qipan[i][m]==qizi){
return false;
}
}
int i=0,j=0;
for(i=k,j=m;i<n&&j<n;i++,j++){ //右下角;
if(qipan[i][j]==qizi){
return false;
}
}
for(i=k,j=m;i>=0&&j>=0;i--,j--){ //左上角;
if(qipan[i][j]==qizi){
return false;
}
}
for(i=k,j=m;i>=0&&j<n;i--,j++){ // 右上角;
if(qipan[i][j]==qizi){
return false;
}
}
for(i=k,j=m;i<n&&j>=0;i++,j--){ //左下角;
if(qipan[i][j]==qizi){
return false;
}
}
return true;
}
/**
* 扫描棋盘是否有n个黑和n个白棋子
*/
public static boolean sum(){
int bainum=0;
int heinum=0;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(qipan[i][j]==bai)
bainum++;
if(qipan[i][j]==hei)
heinum++;
}
}
if(bainum==n&&heinum==n){
num++;
return true;
}
return false;
}
/**
* 递归方法的实现
* @param line 行号
* @param qizi 当前放的棋子
*/
public static void mk(int line , int qizi){
if(line>=n) return ; //如果超出了界限n 就返回
for(int i=0;i<n;i++){
if(jc(line,i,qizi)){
qipan[line][i]=qizi;
count++;
if(count>=n){
if(sum()){
print();
}
int c=count;
count=0;
mk(0,bai); //继续递归
count=c;
}
mk(line+1,qizi);
qipan[line][i]=yes; //回溯
count--;
}
}
}
/**
* 打印当前棋盘
*/
public static void print(){
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
System.out.print(qipan[i][j]+" ");
}
System.out.println();
}
System.out.println(":"+num);
}
public static void main(String args[]){
Scanner scan=new Scanner(System.in);
n=scan.nextInt();
qipan=new int[n][n];
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
qipan[i][j]=scan.nextInt();
}
}
mk(0,hei);
System.out.println(num);
}
}