N皇后问题是指在N*N的棋盘上要摆N个皇后,要求任何两个皇后不同行,不同列,也不在同一条斜线上。给定一个整数n,返回n皇后的摆法有多少种。
n=1,返回1
n=2或者3,返回0
n=8,返回92。
有n行n列的棋盘。首先第一个皇后摆在第一行第一个。然后看有多少种摆法。然后再把第一个皇后摆再第一行第二个,看有多少种摆法,直到第一个皇后放到第一行的最后一个,看有多少种摆法。总摆法就是全部相加。
Java代码如下:
package com.company;
public class NQueens {
public static void main(String[] args) {
System.out.printf("" + num(14));
}
public static int num(int n){
if (n<1){
return 0;
}
//record[0] record[1]表示第i行的皇后放到了第几列。
int[] record = new int[n];
return process(0 , record, n);
}
/*
* 目前来到了第i行
* record[0..i-1]表示之前的皇后位置
* n代表整体一共有多少行
* 返回值是,摆完所有的皇后,合理的摆法有多少种。
* record[0...i-1]的皇后,任何两个皇后一定不共行,不共列,不共斜线。
* */
private static int process(int i, int[] record, int n) {
//终止行
if(i==n){
return 1;
} // this is base case
int res = 0;
//检查第i行的所有的列是否有符合标准的。有的话进入下一行。
for (int j = 0; j<n; j++){
//每一列进行检查,是不是跟之前的皇后同列,同斜线。不用判断是不是共行
if(isValid(record,i,j)){
record[i] = j;
res+=process(i+1,record,n);
}
}
return res;
}
//用于检查,第i行第j列的皇后是不是合理的。和之前的record中的所有皇后做比较
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;
}
}
时间复杂度为O(n^n),可以优化,但是时间复杂度不变,可以降低常数项。方式就是把isValid检查函数用位运算来代替。但是这个不能超过32皇后问题。因为11111111。