搜索刷题1-八皇后

题目来源：[USACO1.5] 八皇后 Checker Challenge - 洛谷

解题思路：这道题用深度优先搜索算法（DFS）来寻找所有可能的解决方案。题目的要求是任意两个皇后都不能处于同一行、同一列或同一对角线上。本题可以定义三个数组b1、b2、b3。

• b1数组用来标记已经放置皇后的列；
• b2数组用来标记主对角线上的占位情况（对于在第x行第y列的皇后，其主对角线的标识是x+y）；
• b3数组用来标记副对角线上的占位情况（对于在第x行第y列的皇后，其副对角线的标识是x-y+n-1，这里加n-1是为了保证索引为正数）。

当一个皇后放置在棋盘上时，它将占据一个列位置、一个主对角线位置和一个副对角线位置，这些位置随后在递归调用中被标记为不可用。一旦所有的皇后都成功放置，就找到了一个解，并且递归开始回溯，撤销皇后的位置和相关的标记，继续尝试新的位置。程序会打印出前三个找到的解，并在程序最后输出总的解的数量。

#include<iostream>
#include<cstdio>

// 定义MAXN为最大的棋盘维度，这里设置为100
#define MAXN 100

using namespace std;

// 全局变量声明
int a[MAXN], ans = 0; // a数组用来存放每个皇后的列位置，ans用来计数解的数量
int b1[MAXN], b2[MAXN], b3[MAXN * 2]; // b1、b2和b3数组用来标记攻击线

// DFS递归函数
void dfs(int x, int n) {
    // 当当前行x超过棋盘大小n时，表示找到一个解，递增解的数量ans
    if (x > n) {
        ans++;
        // 如果解的数量不超过3，则打印当前解
        if (ans <= 3) {
            for (int i = 1; i <= n; i++) {
                cout << a[i] << " "; // 打印皇后的列位置
            }
            cout << endl; // 换行，以便于输出下一个解
        }
        return;
    }
    // 尝试在当前行x放置皇后
    for (int i = 1; i <= n; i++) {
        // 检查第i列是否安全，即不受其他皇后攻击
        if (!b1[i] && !b2[x + i] && !b3[x - i + n - 1]) {
            // 放置皇后，并标记攻击线
            a[x] = i; // 在第x行第i列放置皇后
            b1[i] = b2[x + i] = b3[x - i + n - 1] = 1; // 标记列、主对角线和副对角线被占用
            dfs(x + 1, n); // 递归调用，处理下一行
            // 回溯，撤销皇后的位置，并清除攻击线的标记
            b1[i] = b2[x + i] = b3[x - i + n - 1] = 0;
        }
    }
}

// 主函数
int main() {
    int n; // 棋盘大小
    cin >> n; // 读入棋盘大小
    dfs(1, n); // 从第1行开始DFS
    cout << ans << endl; // 输出解的总数
    return 0;
}