问题分析
N皇后问题是比较经典的问题,它蕴含了回溯的思想,本文章利用回溯算法求解N皇后问题。
算法解析
①本文章是逐行去搜索皇后的位置,每一行逐列检验该位置是否可以安放一个皇后,如果可以安放则记录列号;否则继续检索本行下一列。
for(int i=0;i<queenNum;i++){//一行中,逐列去检索
if(isAble(rowNum,i)){//检验该位置是否可以安放
column[rowNum] = i;//记录列号
findQueen(rowNum+1);//下一行对下一个皇后进行安放
}
}
②当行号数量与皇后数量相同时,认为找到了一种解,解计数器自加1,结束本层递归
if(rowNum==queenNum){
count++;
return;
}
③检索皇后位置完整代码
void findQueen(int rowNum){
if(rowNum==queenNum){
count++;
return;
}
for(int i=0;i<queenNum;i++){//一行中,逐列去检索
if(isAble(rowNum,i)){//检验该位置是否可以安放
column[rowNum] = i;//记录列号
findQueen(rowNum+1);//下一行对下一个皇后进行安放
}
}
}
④ 对于皇后的检索,难点在于对递归与回溯的理解。对于每一次找到一个解,将会结束第n层递归调用,但是第n-1层中的for循环会继续检索本层中未检索的列,这样将会得出新的位置,相应的第n层也会去再次检索出新的位置,从而产生一个新的解。那么对于n层来说,每一层的递归调用都是如此,我们便可以找出所有的解。
⑤安放位置的检验非常简单,在安放第i个皇后时,保证该皇后所在列和所在两条对角线的前i-1行中没有皇后被安放即可
for(int i=0;i<row;i++){
if(column[i]==col||abs(col-column[i]) == abs(i-row)){
return 0;
}
}
return 1;
完整代码
#include <iostream>
#include <stdlib.h>
using namespace std;
int column[100];
int queenNum=0;
int count =0;
int isAble(int row,int col){
for(int i=0;i<row;i++){
if(column[i]==col||abs(col-column[i]) == abs(i-row)){
return 0;
}
}
return 1;
}
void findQueen(int rowNum){
if(rowNum==queenNum){
count++;
return;
}
for(int i=0;i<queenNum;i++){//一行中,逐列去检索
if(isAble(rowNum,i)){//检验该位置是否可以安放
column[rowNum] = i;//记录列号
findQueen(rowNum+1);//下一行对下一个皇后进行安
}
}
}
int main(){
cin>>queenNum;
findQueen(0);
cout<<"种数:"<<count<<endl;
}