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
思路基本上和n皇后问题相似,唯一不同的就是,这次放皇后分为了两部分,即将白皇后和黑皇后分开放;有不懂的可以去看我的n皇后;
package aa;
import java.util.Scanner;
public class 二N皇后问题 {
static int n,count=0; //n表示棋盘的大小,count则为上文提到的计数变量记录放置方法的个数
static int[][] map; //二位数组表示棋盘
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc=new Scanner(System.in);
n=sc.nextInt(); //相关变量的录入
map=new int[n][n];
for(int i=0;i<n;i++) {
for(int j=0;j<n;j++) { //棋盘具体值的录入
map[i][j]=sc.nextInt();
}
}
Put(0,2); //从第1行开始进行白皇后的放置,2代表白,3代表黑
System.out.println(count);
}
private static void Put(int t, int s) { //放置皇后的函数
// TODO Auto-generated method stub
if(t==n) { //进行当前类型皇后的防止数量是否达到要求,即是否到达了棋盘的最后一行
if(s==2)Put(0,3); //如果白黄后放置成功,则进行黑皇后的放置
else count++; //设置方法找到了一种,计数变量进行值加1
return; //当前整体放置过程结束,进行程序的回溯
}
for(int i=0;i<n;i++) { //对每一行的值逐个进行操作
if(map[t][i]!=1)continue; //当前位置是否为1的判断
if(Check(t,i,s)) {
map[t][i]=s; //是否满足题目要求判断,满足赋值
}else continue; //不满足,同一行的下一个位置
Put(t+1,s); //下一行皇后的放置
map[t][i]=1; //回溯发法的关键
}
return;
}
private static boolean Check(int t, int i, int s) { //满足题目要求的判断函数
// TODO Auto-generated method stub
for(int q=t-1;q>=0;q--) {
if(map[q][i]==s)return false;
}
for(int q=t-1,w=i-1;q>=0&&w>=0;q--,w--) {
if(map[q][w]==s)return false;
}
for(int q=t-1,w=i+1;q>=0&&w<=(n-1);q--,w++) {
if(map[q][w]==s)return false;
}
return true;
}
}