回溯算法|八皇后问题

// EightQueen.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

#include <iostream>
#include <vector>

using namespace std;

// 实现在（x,y）放置皇后，对attack数组更新
// x、y表示放置皇后的坐标，二维数组attack表示棋盘是否可以放置皇后
void put_queen(int x, int y, vector<vector<int>>& attack)
{
    // 方向数组，方便后面对8个方向进行标记
    static const int dx[] = { -1,-1,-1,0, 1, 1, 1, 0 };
    static const int dy[] = { -1, 0, 1, 1, 1, 0,-1,-1 };
    attack[x][y] = 1;   // 将皇后位置标记为1

    // 通过两层循环，将皇后可能攻击到的位置进行标记
    for (int i = 1; i < attack.size(); i++) // 从皇后位置向1到n-1个距离延申
    {
        for (int j = 0; j < 8; j++) // 遍历8个方向
        {
            int nx = x + i * dx[j]; // 生成新的位置行
            int ny = y + i * dy[j]; // 生成新的位置列
            // 新位置在棋盘范围内标记为1
            if (nx >= 0 && nx < attack.size() && ny >= 0 && ny < attack.size())
            {
                attack[nx][ny] = 1; // 将新位置标记为1
            }
        }
    }
}

// 回溯法求解N皇后的递归函数
// k表示当前的处理的行
// n表示N皇后问题
// queen存储皇后的位置
// attack标记皇后的攻击位置
// solve存储N皇后的全部解法
void backtrack(int k, int n,vector<string> &queen,
    vector<vector<int>> &attack, vector<vector<string>> &solve)
{
    if (k == n) // 如果当前行等于最后一行则输出皇后的摆放位置
    {
        solve.push_back(queen);
        return;
    }

    // 遍历0至n-1列，在循环中，回溯试探皇后可可知放置的位置
    for (int i = 0; i < n; i++)
    {
        if (attack[k][i] == 0)  // 判断当前第k行第i列是否可以放置皇后
        {
            vector<vector<int>> tmp = attack;   // 备份attack数组
            queen[k][i] = 'Q';  // 标记该位置为‘Q’
            put_queen(k, i, attack);    // 更新attack数组
            backtrack(n + 1, n, queen, attack, solve);  // 递归试探k+1行的皇后放置位置
            attack = tmp;   // 将攻击位置变为原来的方式
            queen[k][i] = '.';  // 取消皇后的放置
        }
    }
}

vector<vector<string>> solveQueens(int n)
{
    vector<vector<string>> solve;   // 存储最后结果
    vector<vector<int>> attack; // 标记皇后的攻击位置
    vector<string> queen;   // 保存皇后位置

    // 使用循环初始化attack和queen数组
    for (int i = 0; i < n; i++)
    {
        attack.push_back((vector<int>()));
        for (int j = 0; j < n; j++)
        {
            attack[i].push_back(0);
        }
        queen.push_back("");
        queen[i].append(n, '.');
    }
    backtrack(0, n, queen, attack, solve);  // 调用backtrack求解N皇后问题
    return solve;
}

int main()
{
    vector<vector<string>> result;
    result = solveQueens(8);

    printf("8皇后共有%d种解法：\n\n", result.size());
    for (int i = 0; i < result.size(); i++)
    {
        printf("解法%d:\n", i+1);
        for (int j = 0; j < result[i].size(); j++)
        {
            printf("%s\n",result[i][j].c_str());
        }
        printf("\n");
    }
    return 0;
}

// 运行程序: Ctrl + F5 或调试 >“开始执行(不调试)”菜单
// 调试程序: F5 或调试 >“开始调试”菜单

// 入门使用技巧: 
//   1. 使用解决方案资源管理器窗口添加/管理文件
//   2. 使用团队资源管理器窗口连接到源代码管理
//   3. 使用输出窗口查看生成输出和其他消息
//   4. 使用错误列表窗口查看错误
//   5. 转到“项目”>“添加新项”以创建新的代码文件，或转到“项目”>“添加现有项”以将现有代码文件添加到项目
//   6. 将来，若要再次打开此项目，请转到“文件”>“打开”>“项目”并选择 .sln 文件