受伤的皇后(蓝桥杯)（DFS）

Question：

题目描述

有一个 n×n 的国际象棋棋盘（n行n列的方格图），请在棋盘中摆放 n 个受伤的国际象棋皇后，要求：

1. 任何两个皇后不在同一行。
2. 任何两个皇后不在同一列。
3. 如果两个皇后在同一条 45 度角的斜线上，这两个皇后之间行号的差值至少为 3 。

请问一共有多少种摆放方案。

输入描述

输入的第一行包含一个整数 nn。

其中，1≤n≤10。

输出描述

输出一个整数，表示答案。

Solve：

标准的DFS板子题，这个题与八皇后唯一不同的就是多了一个对角线差值的限制，判断的时候加上就可以，主对角线用deep-i+n标记，副对角线用deep+i标记

#include<iostream>
#include<cstring>
using namespace std;
bool lie[100],zhu[1000],fu[1000];
int n,cnt = 0,a[11];

void dfs(int deep)
{
    if(deep > n) { cnt++; return; }
    for(int i = 1; i <= n; i++){
        //不在同一行条件必然满足，判断另两个条件
        if(lie[i] == true) continue;
        if((zhu[deep - i + n] == true && (a[deep-1] == i - 1 || a[deep-2] == i - 2) ) ) continue;
        if( (fu[deep + i] == true && (a[deep-1] == i + 1 || a[deep-2] == i + 2) ) ) continue;
        //将皇后放入，继续搜索
        lie[i] = true; zhu[deep - i + n] = true; fu[deep + i] = true;
        a[deep] = i;
        dfs(deep+1);
        lie[i] = false; zhu[deep - i + n] = false; fu[deep + i] = false;
    }
}
int main(void)
{
    memset(lie,false,sizeof(lie));
    memset(zhu,false,sizeof(zhu));
    memset(fu,false,sizeof(fu));
    cin >>n;
    dfs(1);
    cout <<cnt;
    return 0;
}