八皇后—回溯算法、n皇后问题

描述

会下国际象棋的人都很清楚：皇后可以在横、竖、斜线上不限步数地吃掉其他棋子。如何将8个皇后放在棋盘上（有8 * 8个方格），使它们谁也不能被吃掉！这就是著名的八皇后问题。 对于某个满足要求的8皇后的摆放方法，定义一个皇后串a与之对应，即a=b1b2...b8，其中bi为相应摆法中第i行皇后所处的列数。已经知道8皇后问题一共有92组解（即92个不同的皇后串）。 给出一个数b，要求输出第b个串。串的比较是这样的：皇后串x置于皇后串y之前，当且仅当将x视为整数时比y小。

输入描述：

每组测试数据占1行，包括一个正整数b(1 <= b <= 92)

输出描述：

输出有n行，每行输出对应一个输入。输出应是一个正整数，是对应于b的皇后串。

示例1

输入：

1
92

输出：

15863724
84136275

解题思路：

（1）从上往下遍历所有位置是否符合条件，递归遍历；

（2）由于是从上往下，故该行下方的位置不需判断，只需判断同一列，左上和右上的位置即可；

（3）将容器a的数进行排序；

代码：

#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;

vector<int> a;
int Q[8][8]={0};

bool judge(int row,int col){  //判断是否可放皇后；从上往下
    for(int i=0;i<8;i++){  //列 
        if(Q[i][col]==1) return false;
    }
    for(int i=row,j=col;i>=0&&j>=0;i--,j--){ //左上
        if(Q[i][j]==1) return false;
    }

    for(int i=row,j=col;i>=0&&j<8;i--,j++){  //右上
        if(Q[i][j]==1) return false;
    }
    return true;

}

void dfs(int row,int str){  (行数，所求“字符串”（整数）)
    if(row==8){  //行坐标为八时，完成一次放皇后
        a.push_back(str);  //压入a容器中
        return ;  //递归出口
    }else{
        for(int count=0;count<8;count++){  //遍历所有邻居是否可放皇后，列坐标
            if(judge(row,count)){
                Q[row][count]=1;  //可放，把值设为1
                dfs(row+1, str*10+count+1);  //下一行
                Q[row][count]=0;  //回溯
            }
        }
    }
}

int main() {
    dfs(0,0);  //遍历
    int n;
    sort(a.begin(), a.end()); //将a容器中的数据排序
    while (cin >> n) { 
        cout << a[n-1]<< endl;
    }
}