计蒜客2n皇后问题（dfs）

题目链接：点击打开链接

#include<stdio.h>
int map[10][10],b[10],w[10];
int n,num;
void bdfs(int pos)
{
int i;
for(i=0;i<pos-1;i++)
{
int judge=b[pos-1]-b[i];
if(judge==0||judge==pos-1-i||-judge==pos-i-1)
return ;
}
if(pos==n)
{
num++;
return ;
}
for(i=0;i<n;i++)
if(i!=w[pos]&&map[pos][i])
{
b[pos]=i;
bdfs(pos+1);
}
}
void wdfs(int pos)
{
int i;
for(i=0;i<pos-1;i++)
{
int judge=w[pos-1]-w[i];
if(judge==0||judge==pos-1-i||-judge==pos-1-i)
return ;
}
if(pos==n)
{
bdfs(0);
return ;
}
for(i=0;i<n;i++)
if(map[pos][i])
{
w[pos]=i;
wdfs(pos+1);
}
}
int main()
{
int i,j;
scanf("%d",&n);
for(i=0;i<n;i++)
for(j=0;j<n;j++)
scanf("%d",&map[i][j]);
num=0;
wdfs(0);
printf("%d\n",num);
return 0;
}