C++实现n皇后问题

Halo Cooker

已于 2023-05-31 23:12:57 修改

阅读量1k

点赞数

分类专栏：数据结构与算法文章标签： c++ 算法

于 2023-05-30 19:19:22 首次发布

本文链接：https://blog.csdn.net/m0_56350559/article/details/130955254

版权

数据结构与算法专栏收录该内容

5 篇文章 0 订阅

订阅专栏

本文展示了一个C++程序，该程序运用递归和回溯策略来解决经典的N皇后问题。程序初始化标记数组，通过递归函数fn逐步放置皇后，同时避免冲突。SetQueenPos函数更新标记数组以检查冲突。最后，程序输出所有可能的解决方案。

摘要由CSDN通过智能技术生成

我们在数据结构中学到了递归和回溯，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;
            }
        }
    }
}