问题描述:在8×8格的国际象棋上摆放8个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。
代码实现:(具体讲解内容详见代码中的注释)
public class 八皇后 {
static int[] cols = new int[8];
//首先八皇后问题中每一列只能有一个皇后
//所以可以创建一个长度为8的数组,其每一个元素的数值都是相对应的那一列的皇后具体在哪一行
//其次每一行也都只能有一个皇后,所以这个数组的元素应该是0~7之间的不可重复的数
//这样我们在下面的代码中就不用考虑同一行或同一列是否会有两个以上皇后
static int num = 0;//用num记录一共有多少种解法
static void dfs(int arr[],int n) {
//主函数中会将arr这个参数设置成{0,1,2,3,4,5,6,7}
//这个函数用回溯算法求arr的全排列,然后把排列赋值给八个皇后的位置信息cols
//不懂回溯算法的可以参考https://blog.csdn.net/qq_45724216/article/details/113996838的第二块代码片
//然后用isLegal函数判断八皇后是否满足不处于同一斜线上,不用判断是否出于同一列或同一行
//原因详见定义整形数组cols的哪一行代码下面的注释
if(n==arr.length) {
cols=arr;
if(isLegal()) {
//isLegal和printChessBoard都是自定义函数
//isLegal判断不处于同一斜线上
//printChessBoard打印棋盘
printChessBoard();
}
}
for(int i=n;i<arr.length;i++) {
swap(arr,i,n);
//swap是自定义函数,调换数组中第i个元素和第j个元素
dfs(arr,n+1);
swap(arr,n,i);
}
}
static void swap(int[] arr,int i,int j) {
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
static void printChessBoard() {
for(int i=0;i<8;i++) {
for(int j=0;j<8;j++) {
if(i == cols[j]) {
System.out.print("0 ");
}else {
System.out.print("+ ");
}
}
System.out.println();
}
num++;
System.out.println("八皇后第"+num+"种解法\n"+"\n");
}
static boolean isLegal() {
//这个函数比较抽象,我不多赘述,看不懂就看下面的图解
boolean flag = true;
for(int i=0;i<8;i++) {
for(int j=i+1;j<8;j++) {
if((j-i)==Math.abs(cols[j]-cols[i]))flag = false;
}
}
return flag;
}
public static void main(String[] args) {
int[] arr = {0,1,2,3,4,5,6,7};
dfs(arr,0);
System.out.println("八皇后问题一共有"+num+"种解法");
}
}
由图可以看到,对于第i列和第j列的皇后
如果cols[i]-cols[j]等于它们的索引值之差或与索引值之差成相反数
皇后位置不合法
而布尔型数据flag表示皇后位置合不合法
所以对于第i列和第j列的皇后
if((j-i)==Math.abs(cols[j]-cols[i]))flag = false;
然后将这行语句镶入for循环,对每一个皇后开始遍历。
对于第i列的皇后,让她与第i+1~7位皇后进行比较即可