N皇后问题(深搜 重做详解)

P1219

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=100;
//6x6  棋盘被放置，使得每行每列只有一个，每对角线线至多有一个
/*
解决行冲突，规定一行只放一个皇后
列：定义一个一个标记数组flag[]置为被占领状态
对角线：对角线有两个方向，当第n行第col列被占领后，要同时把 上下对角线都只为占领状态
*/
int a[N],b[N],c[N],d[N];
// a 分别表示行
//b 列
//c 左上到右下(下对角线)
//d 左下到右上（上对角线）
/*
1.初始化（清楚棋盘）
2.循环八次：
 放置一个皇后；
 检查是否满足条件，满足就登记皇后位置
 不满足就退回
3.直到放到最后一个皇后

4.递归终止条件找到最后一个皇后的位置

算法流程：
1.初始化
2.从col列开始摆放第n个皇后后（因为这样便可以符合每一个竖列一个皇后的要求）
先看第n行

3.place[] 第n行皇后所占的位置的列号 主要用于输出结果
  flag[] bool数组col列是否可放
  d1上对角线:行减去列+7（转换成正数）
  d2下对角线:列加行

程序思路：
1.是否可放：
2.列是否被占领，d1[n+col+7],d2[n+col]都为真
3.放置皇后：
place[n]:col
flag[col]:0列被占领
d1[n-col+7]=0
d2[n+col]=0
4.删除元素：（回溯）
flag[col]=1;
d1[n-col+7]=1
d2[n+col]=1
*/
int total;
int n;
int print()
{
    if(total<=2)//保证只输出前三个解，如果解超出三个就不在输出，但后面的total还需要继续
    {
        for(int k=1; k<=n; k++)
        {
            cout<<a[k]<<" ";
        }
        cout <<endl;
    }
    total++;
}
void queen(int i)//搜索与回溯主体
{
    if(i>n)
    {
        print();//输出函数，自己写
        return ;
    }
    else
    {
        for(int j=1; j<=n; j++) //尝试可能的位置
        {
            if((!b[j])&&(!c[i+j])&&(!d[i-j+n]))//如果没有皇后占领，执行以下程序
            {
                a[i]=j;//标记i排是第j个
                b[j]=1;//宣布占领
                c[i+j]=1;
                d[i-j+n]=1;

                queen(i+1);//进行下一步搜搜，下一个皇后
                
                b[j]=0;
                c[i+j]=0;
                d[i-j+n]=0;
                //回溯(可以将递归模拟一下，就清楚了)
                //这里开始执行第一次的时候是已经print()了一种方案，回溯是为了将for循环执行完寻找
                //寻找其他方案
            }
        }
    }
}
int main()
{
    cin>>n;
    queen(1);
    cout<<total<<endl;
    return 0;
}

分享一个up主的 视频详解