[回溯搜索] 八皇后

今天来给大家讲八皇后这道题目

题目传送门(洛谷)
一个如下的 $6\times 6$ 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出，解按字典顺序排列。
请输出前 33 个解。最后一行是解的总个数。

输入格式
一行一个正整数 nn，表示棋盘是 n \times nn×n 大小的。

输出格式
前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

输入输出样例
输入 #1
6
输出 #1复制
2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4
说明/提示
【数据范围】
对于100% 的数据，$6\le n\le 13$

题目翻译来自NOCOW。

USACO Training Section 1.5

思路

这个题目嘛说时候也没什么可以说的 就是一个简简单单的回溯搜索
注意一下标记和回溯的清空场地
其他的就看代码的注释吧

代码

#include<iostream>
using namespace std;
#define maxn 1005
int a[maxn],b[maxn],c[maxn],d[maxn];
//a数组表示的是行；
//b数组表示的是列；
//c表示的是左下到右上的对角线；
//d表示的是左上到右下的对角线；
int anscnt;//解的数量
int n;
int print()
{
    if(anscnt<=2)//保证只输出前三个解
    {
        for(int k=1;k<=n;k++)
        cout<<a[k]<<" ";
        cout<<endl;
    }
    anscnt++;
}//输出函数
void dfs(int dep)
{
    if(dep>n)
    {
        print();
        return;
    }
    for(int i=1;i<=n;++i)
    {
        if(b[i]==0&&c[dep+i]==0&&d[dep-i+n]==0)//判断是否有皇后占领
        {
            a[dep]=i;//标记dep排是第i个
            b[i]=1;//占领纵列
            c[dep+i]=1;//占领第一条对角线
            d[dep-i+n]=1;//占领第二条对角线
            dfs(dep+1);
            b[i]=0,c[dep+i]=0,d[dep-i+n]=0;
           //清空所有标记 回溯搜索的关键
        }
    }
}
int main()
{
    cin>>n;
    dfs(1);//从第一个皇后开始
    cout<<anscnt<<endl;//输出解的个数
}