利用二维数组vis[2][]直接判断当前尝试的皇后所在的列和两个对角线是否有其他皇后。注意主对角线标志y-x可能为负,存取时要加上n。
vis数组的使用。
vis数组的确切含义是表示已经放置的皇后占据了哪些列、主对角线和副对角线。
另外在调用前一定要把vis数组清空
#include<iostream>
#include<stdio.h>#include<algorithm>
#include<math.h>
#include<string.h>
#include<string>
#include<set>
#include<sstream>
using namespace std;
int vis[3][20],c[20];
int n;
int ans=0;
void dfs(int row)
{
if(row==n)
{
ans++;
return ;
}
for(int col=0;col<n;col++)
{
if(!vis[0][col]&&!vis[1][row+col]&&!vis[2][row-col+n])// 利用二维数组直接判断
{
c[row]=col;// 如果不打印解,整个c数组都可以省略
vis[0][col]=vis[1][row+col]=vis[2][row-col+n]=1;
dfs(row+1);
vis[0][col]=vis[1][row+col]=vis[2][row-col+n]=0;
}
}
}
int main()
{
while(~scanf("%d",&n))
{
memset(vis,0,sizeof(vis));
dfs(0);printf("%d\n",ans);
}
return 0;
}