回溯法核心代码:
void generateP(index)
{
if(符合边界条件){
计数,返回上一级;
}
for(枚举情况){
if(能进行到下一代){
flag=true; //初始化
generateP(index+1);
flag=false;
}
}
}
n皇后问题回溯法
/*n皇后问题回溯法*/
#include "stdlib.h"
#include "cstdio"
#define MAX 1000
bool hash[MAX];
int p[MAX];
int count=0;
int n=0;
void print(){
int i,j;
for(i=1;i<=n;i++){
for(j=1;j<=n;j++){
if(p[i]==j){
printf("%d",1);printf(" ");
}
else{
printf("%d",0);printf(" ");
}
printf("\n");
}
}
}
int generate(int index){//一个generate是一行
if(index==n+1){ //最后,count++
//print();
count++;
return 0;
}
for(int i=1;i<=n;i++){
if(hash[i]==false){
bool flag=true;
for(int pre=1;pre<index;pre++){
if(abs(index-pre)==abs(i-p[pre])){
flag=false;
break;
}
}
if(flag){
p[index]=i;
hash[i]=true;
generate(index+1);
hash[i]=false;
}
}
}
return 0;
}
int main()
{
scanf("%d",&n);
generate(1);
printf("%d",count);
system("pause");
return 0;
}