八皇后问题,用回溯法解决:
#include<iostream>
#include<vector>
using namespace std;
#define N 8
#define S 3
#define Q 1
void print(char a[N][N],int count){
cout<<"-->"<<count<<endl;
for(int i=0;i<N;i++){
for(int j=0;j<N;j++){
cout<<a[i][j]<<" ";
}
cout<<endl;
}
}
bool valid(char a[N][N],int i,int j){
for(int k=0;k<N;k++){
if(k!=j&&a[i][k]==Q)
return false;
}
for(int k=0;k<N;k++){
if(k!=i&&a[k][j]==Q)
return false;
}
int x=i,y=j;
while(x>0&&y>0){
if(a[--x][--y]==Q)
return false;
}
x=i,y=j;
while(x<N-1&&y<N-1){
if(a[++x][++y]==Q)
return false;
}
x=i,y=j;
while(x>0&&y<N-1){
if(a[--x][++y]==Q)
return false;
}
x=i,y=j;
while(x<N-1&&y>0){
if(a[++x][--y]==Q)
return false;
}
return true;
}
void trial(char a[N][N],int i,int &count){
if(i==N){
count++;
print(a,count);
}else{
for(int j=0;j<N;j++){
a[i][j]=Q;
if(valid(a,i,j))
trial(a,i+1,count);
a[i][j]=S;
}
}
}
int main(){
char a[N][N];
//init
for(int i=0;i<N;i++){
for(int j=0;j<N;j++){
a[i][j]=S;
}
}
int count=0;
trial(a,0,count);
return 0;
}