n皇后问题—回溯法 C++实现

#include <iostream>
#include <vector>
using namespace std;
bool isLegal(int row,int col,vector<string> &v,int n)
{
for(int i=0;i<row;++i)
if(v[i][col]=='Q') return false;
for(int i=row-1,j=col-1;i>=0&&j>=0;--i,--j)
if(v[i][j]=='Q') return false;
for(int i=row-1,j=col+1;i>=0&&j<n;--i,++j)
if(v[i][j]=='Q') return false;
return true;
}
void snqHelp(vector<vector<string>> &res,vector<string> &v,int row,int n)
{
if(row==n)
{
res.push_back(v);
return;
}
for(int col=0;col<n;++col)
{
if(isLegal(row,col,v,n))
{
v[row][col]='Q';
snqHelp(res,v,row+1,n);
v[row][col]='.';
}
}
}
vector<vector<string>> solveNQueens(int n)
{
vector<vector<string>> res;
vector<string> v(n,string(n,'.'));
snqHelp(res,v,0,n);
return res;
}
int main(int argc, char const *argv[])
{
int n;
cout<<"input n :";
cin>>n;
vector<vector<string>> res=solveNQueens(n);
for(auto vstr:res)
{
for(auto x:vstr)
cout<<x<<endl;
string line(2*n,'-');
cout<<line<<endl;
}
cout<<"Total solution :"<<res.size()<<endl;
return 0;
}