N后问题回溯

#include"iostream"
#include"vector"
#include"string"
using namespace std;
//由于在x,y坐标上放入皇后,8个方向都不能有其他皇后,所以将这8个方向对应的坐标置1
void put_down_the_queen(int x,int y,vector<vector<int> > &mark)
{
    static const int dx[]={-1,1,0,0,-1,1,1,-1};
    static const int dy[]={0,0,-1,1,1,1,-1,-1};
    mark[x][y] = 1;
    for(int i=1;i<mark.size();i++)
       for(int j=0;j<8;j++){
          int new_x = x + i*dx[j];
          int new_y = y + i*dy[j];
          if(new_x>=0&&new_y>=0&&new_x<mark.size()&&new_y<mark.size())
            mark[new_x][new_y] = 1;
    }
}
//k表示行,n表示列,location作为放置皇后Q的棋盘,mark作为放入皇后后的坐标(1表示不能放,0可以)
//result作为保存最后的生成结果
void generate(int k,int n,
              vector<string>  &location,
              vector<vector<string> > &result,
              vector<vector<int> > &mark)
{
    if(k==n){
        result.push_back(location);
        return;
    }
    for(int i=0;i<n;i++){
        if(mark[k][i]==0){
            vector<vector<int> > temp_mark = mark;
            location[k][i] = 'Q';
            put_down_the_queen(k,i,mark);
            generate(k+1,n,location,result,mark);
            mark = temp_mark;
            location[k][i] = '.';
        }
    }



}
int main()
{
    int n;
    cin>>n;
    vector<vector<string> >   result;
    vector<string>   location;
    vector<vector<int> >    mark(n,vector<int>(n,0));
    for(int i=0;i<n;i++)
    {
        location.push_back("");
        location[i].append(n,'.');
    }
    generate(0,n,location,result,mark);
    for(int i=0;i<result.size();i++)
    {
        for(int j=0;j<n;j++)
    {
        cout<<result[i][j];
        cout<<endl;
    }
    cout<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/jd_457619512/article/details/88824485