N皇后问题
题目:N皇后问题是8皇后问题的进阶版,但是处理方式一样,由十九世纪著名的数学家高斯于1850年提出,问题是:在8×8的棋盘上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上。可以把八皇后问题扩展到n皇后问题,即在n×n的棋盘上摆放n个皇后,使任意两个皇后都不能处于同一行、同一列或同一斜线上。
按照题目的意思,大家可以先看一下4皇后问题。(因为2皇后和3皇后问题是没有解的)
由上图,大家也可以理解该题的解法,就是一层一层判断出皇后出现的位置是否符合逻辑,大家可能想到的第一个方法就是使用for循环进行处理,比如4皇后问题,可以使用四个for循环进行求解,但是我们处理N皇后问题不方便,于是我们可以使用回溯的方法进行处理。
回溯和循环的时间复杂度相同,对于该题O(n^n)。
可以使用record[]数组存放合法的置放皇后条件,由于一行只让存放一个皇后,所以我们可以将record[[0,0],[1,1],[2,2]](在第一列第一个,第二列第二个,第三列第三个存放皇后),转换为record[0,1,2],让其数组的下标作为皇后的行数。
根据上图,第一列第四列有一个皇后,绿色的线的区域就是不合法区域,别的皇后不可以放在这些位置,判断是否冲突就是需要满足三个条件:
- 不在同一列
- 不在同一行
- 不在对角线位置
于是对于这样我们可以得到判断条件:
- 由于我们一行只会放一个皇后,条件1 就不用写条件
- 对于条件2,我们使得record数组之中的元素不同
- 对于条件3,我们可以先设有两个皇后为a[x,y],a[i,j],|i-x|==|j-y|
对于条件3的解释(懂得可以不用看这部分):
只看对角线这一部分,我们可以得到对角线部分的横纵坐标之差的绝对值相同,所以只要满足|i-x|==|j-y|的反条件,就可以满足条件3。
结束条件就是当循环到最后一行时,返回正确的解数量,就可以得到答案。
最终得到代码:
import java.util.Scanner;
public class Nqueens {
public static int num1(int n) {
if(n<1) {
return 0;
}
int[] record=new int[n];
return process1(0,record,n);
}
//从i行开始放置皇后,每一列都要尝试
//然后向下遍历,但是要判断在下面放置皇后的时候不可以与之前的冲突
//record是一个一维数组,record[x]=y表示,在第x行,皇后在y列上
//因为一行只能放一个皇后,所以用下标就可以表示皇后的行
//回溯结束的条件,就是i==n,因为由于不符合条件就会返回false,所以就可以直接返回最后一次的结果(里面的都是合法的)
public static int process1(int i,int[] record,int n) {
if(i==n) {
return 1;
}
int res=0;
//皇后放在i行j列
for(int j=0;j<n;j++) {
if(isValid(record,i,j)) {
record[i]=j;
res+=process1(i+1, record, n);
}
}
return res;
}
//判断是否冲突的函数
//两个条件:1.不能在相同的列数
// 2,不能在对角线上
private static boolean isValid(int[] record, int i, int j) {
for(int k=0;k<i;k++) {
if(j==record[k]||Math.abs(record[k]-j)==Math.abs(i-k)) {
return false;
}
}
return true;
}
//main
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
System.out.println(num1(n));
}
}
欢迎大家指正!