java N皇后问题 回溯法
1.什么是N皇后问题
N皇后问题由八皇后问题衍生而来,那么什么又叫作八皇后问题呢?
八皇后问题,是一个古老而著名的问题,是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯·贝瑟尔于1848年提出:在8×8格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。由此拓展出n皇后问题,即在一个n×n的棋盘里放置n个皇后,要求这n个皇后两两之间互相都不“冲突”,共有多少种放置方案
N皇后问题是为了统计所有解法,解空间的大小为2n。
2.解决N皇后问题的算法思路
回溯法思想:在问题的解空间中,按深度优先遍历策略,从根节点出发搜索解空间树。算法搜索至解空间的任意一个节点时,先判断该节点是否包含问题的解。如果肯定不包含,跳过对以该节点为根的子树的搜索,逐层向其祖先节点回溯,否则进入该子树,继续深度优先搜索。
回溯法解问题的所有解时,必须回溯到根节点,且根节点的所有子树都被搜索后才结束。
回溯的基本步骤:
1.定义问题的解空间
2.确定易于搜索的解空间结构
3.以深度优先搜索的策略搜索解空间,并在搜索过程中用剪枝函数避免无效搜索
本题利用递归的办法,采用回溯法进行深度遍历,符合条件的继续向下遍历,不符合条件的,回溯到之前的状态,问题的出口为深度大于等于n时,就是一种符合条件的解法,里面剪枝操作可以减少许多种不必要的考虑,从而使整个问题的运算效率提高,但是由于解空间的大小和这个问题的要求,导致这个办法仍是指数级的,对于n较大的数运行时间过长。
3.代码实现
package math;
//n皇后(回溯法)
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Scanner;
public class test4 {
public ArrayList<ArrayList<String>> solveNQ(int n){
ArrayList<ArrayList<String>> res = new ArrayList<>();
char[][] board = new char[n][n];
init(board);
backtrack(res,board,0);
return res;
}
private void init(char[][] board) {
for(int i=0;i<board.length;i++) {
Arrays.fill(board[i],'.');
}
}
//核心递归函数
private void backtrack(ArrayList<ArrayList<String>> res,char[][] board,int row) {
if(row==board.length) {
res.add(generate(board));
return;
}//遍历行到达最底层结束
for(int j=0;j<board.length;j++) {
if(isvalid(board,row,j)) {
board[row][j]='Q';
backtrack(res,board,row+1);//递归调用下一行
board[row][j]='.';//回溯
}
}
}
private boolean isvalid(char[][] board,int row,int col) {
for(int i=0;i<row;i++) {
if(board[i][col]=='Q')return false;
}
for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--) {
if(board[i][j]=='Q')return false;
}
for(int i=row-1,j=col+1;i>=0&&j<board.length;i--,j++) {
if(board[i][j]=='Q')return false;
}
return true;
}
private ArrayList<String> generate(char[][] board){
ArrayList<String> list = new ArrayList<String>();
for(char[] row:board) {
StringBuilder sb=new StringBuilder();
for(char c:row) {
sb.append(c);
}
list.add(sb.toString());
}
return list;
}
public static void main(String[] args) {
test4 t=new test4();
System.out.println("请输入皇后个数:");
Scanner scanner=new Scanner(System.in);
int n=scanner.nextInt();
ArrayList<ArrayList<String>> res= t.solveNQ(n);
for(int i=0;i<res.size();i++) {
System.out.println(res.get(i));
}
System.out.println("共有解法数:"+res.size());
}
}
4.实验结果
省略中间部分,只截取部分结果