2*n皇后的放置
描述
给定一个n*n的棋盘,棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后,使任意的两个黑皇后都不在同一行、同一列或同一条对角线上,任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法?n小于等于8
输入
输入的第一行为一个整数n,表示棋盘的大小。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,如果一个整数为0,表示对应的位置不可以放皇后。
输出
输出一个整数,表示总共有多少种放法。
样例输入
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
2
0
解题思路
与普通八皇后放置问题解法相同,先放黑皇后,后放白皇后,进行两次深度搜索回溯就可以了。
注意:在放置白皇后时要判断该位置有没有已经放了黑皇后。
代码
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int map1[10][10];
int dir[10][10];
int a[50],b[50],c[50],d[50],bb[50],cc[50],dd[50],aa[50];
int n;
int tal=0;
void dfs2(int i);
void dfs1(int i)
{
if(i>n)
{
dfs2(1);
}
else
for(int j=1;j<=n;j++)
{
if(!b[j]&&(!c[i+j])&&(!d[i-j+n])&&map1[i][j])
{
a[i]=j;
b[j]=1;
c[j+i]=1;
d[i-j+n]=1;
dir[i][j]=1;
dfs1(i+1);
b[j]=0;
c[j+i]=0;
d[i-j+n]=0;
}
}
}
void dfs2(int i)
{
if(i>n)
{
tal++;
return;
}
else
for(int j=1;j<=n;j++)
{
if(a[i]==j)
continue;
if(!bb[j]&&(!cc[i+j])&&(!dd[i-j+n])&&map1[i][j]&&(a[i]!=j))
{
aa[i]=j;
bb[j]=2;
cc[j+i]=2;
dd[i-j+n]=2;
dfs2(i+1);
bb[j]=0;
cc[j+i]=0;
dd[i-j+n]=0;
}
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",&map1[i][j]);
dfs1(1);
printf("%d\n",tal);
}