问题描述:
在N*N(N<=10)的棋盘上放N个皇后,使得她们不能相互攻击。两个皇后能相互攻击当且仅当它们在同一行,或者同一列,或者同一条对角线上。找出一共有多少种放置方法。
关于如何判断棋子的斜线方向是否存在棋子可参考:https://www.cnblogs.com/Peper/p/7966253.html
我这里采用的是用两个bool数组来分别存储左右对角线的编号。
这里以8皇后为例:
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
int ans=0;
bool visit[10],col1[20],col2[20];//col1代表左对角线(r-i+8是为了防止数组下标出现负号),col2代表右对角线( 数组下标r+i)
bool check(int r,int i)
{
return !visit[i]&&!col1[r+i]&&!col2[r-i+8];
}
void dfs(int r){
if(r==8)
{
ans++;
return;
}
for(int i=0;i<8;i++)
{
if(check(r,i))
{
visit[i]=col1[r+i]=col2[r-i+8]=true;
dfs(r+1);
visit[i]=col1[r+i]=col2[r-i+8]=false;
}
}
}
int main()
{
dfs(0);
cout<<ans<<endl;
return 0;
}