【蓝桥杯】BASIC27 2n皇后问题

感觉自己的智商在地上摩擦，看了题目一头雾水，看了正确代码还花了一个多小时研究人家为啥这么写23333

代码的逻辑是这样的：

 

不是很流程图，主要是想把代码逻辑写下来，不然改天自己又看不懂了

有个小知识点：主对角线行列数之差相等，副对角线行列数之和相等

 代码参考的 2n皇后问题【dfs】 - 悠悠呦~ - 博客园

我加了些注释：

#include <iostream>
#include <math.h>
#include <stdio.h>
#include <cstring>

using namespace std;

/*
2n皇后问题
问题描述
　　给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、
同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。
输入格式
　　输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。
输出格式
　　输出一个整数，表示总共有多少种放法。
*/
/*
回溯法：走一步试试，行就继续前进，不行就退回一步
*/

//abs()绝对值函数
//define的作用是省事，如果直接用memset函数需要写三个参数
#define mem(a,b) memset(a,b,sizeof(a));
int n; //行列数，用于最初的输入
int mmap[10][10]; //地图，用于最初的输入
int black[10],white[10]; //黑皇后数组，白皇后数组，black[x]=i，说明是第x行i列。序号代表行，值代表列，仅用于判断是不是可以放在这里
int visit[10][10]; //有没有放皇后（不分黑白，只看有没有）
int meth; //几种方法

void init() //初始化函数
{
    meth=0;
    //memset(black,0,sizeof(black));
    mem(black,0)
    mem(white,0);
    mem(mmap,0);
    mem(visit,0);
}

bool juge(int arr[],int x) //判断是否可以放在这里的函数
{
    if(mmap[x][arr[x]]==0 || visit[x][arr[x]]==1)
        //最初输入的矩阵不允许在此处放皇后 || 这里已经放了皇后了
        return false;
    for(int i=1;i<x;i++) //只需要遍历之前的行，还没开始放的行不用管
    {
        if(arr[i]==arr[x] || abs(i-x)==abs(arr[x]-arr[i]))
        //if(arr[i]==arr[x] || abs(i-arr[i])==abs(arr[x]-x))
        //这一列已经有皇后了 || 这个主对角线上已经有皇后了
        //数组arr（数组black、white同）下标代表行，值代表列，arr[x]=放在了第x行哪一列，如果用i遍历，发现之前某行已经在这一列放了黑/白皇后，则不行
            return false;
    }
    return true;
}

void dbs_w(int x) //放白皇后的函数
{
    if(x==n+1)
    {
        meth++; //白皇后放完了，方法+1
        return; //返回black()
    }
    for(int i=1;i<=n;i++)
    {
        white[x]=i;
        if(juge(white,x))
        {
            visit[x][i]==1;
            dbs_w(x+1);
            visit[x][i]=0;
        }
    }
}

void dbs_b(int x) //放黑皇后的函数
{
    if(x==n+1) //黑皇后放完了，去放白皇后
    {
        dbs_w(1);
        return; //都放完了，返回main()
    }
    for(int i=1;i<=n;i++) //没放完，在这一行找地方放，从第一个开始挨个试
    {
        black[x]=i; //先放再试
        //这里只用判断是不是到达第n+1行就能判断是否放了n个黑皇后，不用另外用一个变量记录放的黑皇后的数量
        //因为如果某一行没有放黑皇后，那么它根本就不能够向下一行搜索
        if(juge(black,x)) //判断是不是可以放在这里
        {
            visit[x][i]=1; //可以，真正放上，以此为前提往下走
            dbs_b(x+1); //放下一行
            visit[x][i]=0;  //走不下去，回退一步，这个不放了
        }
    }
}

int main()
{
    int i,k;
    init(); //初始化

    cin>>n; //输入，这是个几行几列的矩阵
    for(i=1;i<=n;i++)
    {
        for(k=1;k<=n;k++)
        {
            cin>>mmap[i][k];
        }
    }
    dbs_b(1); //调用黑皇后函数，从第一行开始遍历尝试
    cout<<meth;//输出最后的方法数
    return 0;
}

有个问题，juge()循环函数里的第二个判断条件，相当于 r1-r2==c1-c2 ，理论上是等价于 r1-c1==r2-c2的，但是改了之后就不对了，俺也不知道为啥，捉一只看到这的大佬指点迷津~