思路:按照行来dfs,同时使用col,dg,udg三个数组来记录列,对角线和反对角线是否有棋子。
当我们的u,也就是棋子数等于n的时候,就是我们把棋子安放完成,返回即可。
这里我们的u也同时是行数,这里i则可以理解为列数,从0到n-1列,也就是n列。
我们可以发现一个规律,一条对角线上面的x,y坐标之和相同,例如(1,3),(2,2)或者(2,3)(3,2)。
那么我们把每个点的对应对角线就是dg[u+i]。
这样看来对角线像从左上角依次向下,而反对角线则是从右上角开始。
我们可以发现我们用(i-u)也就是列数减去行数可以表示反对角线的下标,因为这可能会导致为负数,所以我们来加上一个n杜绝这种可能。
最后我们搜索过之后要记得恢复状态,不影响后面的搜索。
代码实现:
#include<bits/stdc++.h>
using namespace std;
const int N = 30;
bool col[N],dg[N],udg[N];
int n,ans=0;
void dfs(int u){
if(u == n){
ans ++;
return;
}
for(int i = 0; i < n; i ++){
if (!col[i] && !dg[u + i] && !udg[n - u + i]){
col[i] = dg[u + i] = udg[n - u + i] = true;//列,双对角线不合格
dfs(u + 1);//搜索下一行
col[i] = dg[u + i] = udg[n - u + i] = false;
}
}
}
int main(){
cin >> n;
dfs(0);
cout << ans;
return 0;
}