题目描述
-
在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。 -
Input
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。 -
Output
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
解析
- 采用”试探法”:即尝试每个位置是否能放,采用dfs思想,并根据要求即:(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。)进行剪枝。
代码
static int count=0;
public static void main(String[] args) {
Scanner scanner=new Scanner(System.in);
int N=scanner.nextInt();
int[][] res=new int[N][N];
//从第0行开始深搜
dfs(res,0);
System.out.println(count);
}
private static void dfs(int[][] res, int row) {
//当左最后一行扫描完以后,退出循环
if(row==res.length) {
count++;
return;
}
for (int col =0; col < res.length; col++) {
if(check(res,row,col)) {
res[row][col]=1;
dfs(res, row+1);
//一条路走到黑之后或者出现不能继续前进的状况时,进行下一种情况的深搜,即回到搜索前的情况
res[row][col]=0;
}
}
}
private static boolean check(int[][] res, int row, int col) {
//扫描当前这一行
for (int i = 0; i < res.length; i++) {
if(res[i][col]==1) return false;
}
//扫描当前这一列
for (int j = 0; j < res[0].length; j++) {
if (res[row][j]==1) {
return false;
}
}
//扫描对角线
for (int i = 0; i < res.length; i++) {
for (int j = 0; j < res[0].length; j++) {
if(i+j==row+col) {
if(res[i][j]==1) {
return false;
}
}
if(i-j==row-col) {
if(res[i][j]==1) {
return false;
}
}
}
}
return true;
}