我们在数据结构中学到了递归和回溯,N后问题就是这种思想的最好体现.
#include<iostream>
#include <vector>
#include<string>
//1.存储最终结果
//2.存储每次结果
//3.存储标记数组
//4.方向数组
using namespace std;
void fn(int k, int n, vector<vector<string>>& ret, vector<string>& temp,vector<vector<int>>& visited);//传入的参数依次是当前皇后个数,所有皇后的个数,纪录所有成功结果的集合,当前结果的类型,当前的状态数组
void SetQueenPos(int n,int x, int y, vector<vector<int>>& visited);//这个函数用来更新标记数组
int main() {
vector<vector<string>> ret;//储存所有结果
vector<vector<int>> visited;//当前的标记数组
vector<string> temp;//当前的结果
int i, j;//用来打印的变量
int n = 16;//皇后的个数
for (i = 0; i < n; ++i) //这个for循环是用来初始化标记数组,当前状态
{
visited.push_back(vector<int>());
for (j = 0; j < n; ++j)
{
visited[i].push_back(0);
}
temp.push_back("");
temp[i].append(n, '*');
}
fn(0, n, ret, temp, visited);//开始进入寻找解的函数
for (i = 0; i < ret.size(); ++i) {//这个for循环是打印最后的解
cout << "----------- -------------" << i + 1 << "------" << endl;
for (j = 0; j< ret[i].size(); ++j) {
cout << ret[i][j].c_str() << endl;
}
}
}
void fn(int k, int n, vector<vector<string>>& ret, vector<string>& temp,
vector<vector<int>>& visited) {
if (k == n) {
ret.push_back(temp); return;
}
for (int i = 0; i < n; i++) {
if (visited[k][i] == 0) {
vector<vector<int>> t_visited = visited;
temp[k][i] = 'q';
SetQueenPos(n,k, i, visited);
fn(k + 1, n, ret,temp, visited);
visited = t_visited;
temp[k][i] = '*';
}
}
}
void SetQueenPos(int n,int x, int y, vector<vector<int>>& visited) {
int newx[] = { -1,1,0,0,-1,-1,1,1 };
int newy[] = { 0,0,-1,1,-1,1,-1,1 };
for (int i = 1; i < visited.size(); ++i) {
for (int j = 0; j < 8; ++j) {
int dx = i * newx[j] + x;
int dy = i * newy[j] + y;
if (dx >= 0 && dx < visited.size() && dy >= 0 && dy < visited.size()) {
visited[dx][dy] = 1;
}
}
}
}