参考链接:https://www.jianshu.com/p/65c8c60b83b8
八皇后问题的主要思想是使用回溯法。我们已经知道,八个皇后分别在每一行, 而所在的列是需要我们用递归来进行寻找的。
#include<bits/stdc++.h>
using namespace std;
bool isOk(vector<vector<int>>& flag, int row, int col ){
for(int i = 0; i < row; i++){ // 判断 所在列是否有其他皇后
if(flag[i][col]==1){
return false;
}
}
for(int i = row-1, j = col-1; i>=0 && j>=0 ; i--,j--){ // 判断左上对角线是否有其他皇后
if(flag[i][j]==1){
return false;
}
}
for(int i = row-1, j = col+1; i>=0 && j<8;i--,j++){ // 判断右上对角线是否有其他皇后
if(flag[i][j]==1){
return false;
}
}
return true;
}
// 依次寻找每一个皇后合适的位置
void findQueen(int & ans, vector<vector<int>>& flag, int row){
if(row>7){ // 此时,所有皇后都找到了其合适的位置, ans++
ans++;
return;
}
for(int i=0;i<8;i++){ // 依次遍历当前皇后放置各个列的情形
if(isOk(flag,row,i)){ // 如果当前位置可以放皇后,那么递归下去,寻找下一个皇后的可行位置
flag[row][i] = 1; // 标识该位置已放皇后
findQueen(ans,flag,row+1);
flag[row][i] = 0; // 需要置0
}
}
// return ;
}
int main(){
int ans = 0;
vector<vector<int>> flag(8,vector<int>(8,0));
findQueen(ans,flag,0);
cout<<ans<<endl;
return 0;
}