给定⼀个n*n的棋盘,棋盘中有⼀些位置不能放皇后。现在要向棋盘中放⼊n个⿊皇后和n个⽩皇后,
使任意的两个⿊皇后都不在同⼀⾏、同⼀列或同⼀条对⻆线上,任意的两个⽩皇后都不在同⼀⾏、同
⼀列或同⼀条对⻆线上。问总共有多少种放法?n⼩于等于8。
接下来n⾏,每⾏n个0或1的整数,如果⼀个整数为1,表示对应的位置可以放皇后,如果⼀个整数为
0,表示对应的位置不可以放皇后。
输出格式
输出⼀个整数,表示总共有多少种放法。
样例输⼊
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
2
样例输⼊
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
0
思路:两次dfs
#include<iostream>
#include<math.h>
using namespace std;
//解题思路:在n皇后弄懂后,这题就是黑白皇后分开处理,可以选择先填白皇后,再填黑皇后
int wmap[9]={0};//白皇后位置
int bmap[9]={0};//黑皇后位置
int v[9][9];//存储初始条件
int n;
int ans=0;
bool wisok(int row,int col)//函数作用:判断白皇后,同一列,同一对角线不能相等
{
for(int i=0;i<row;i++)
if(wmap[i]==col||fabs(wmap[i]-col)==row-i)
return false;
return true;
}
bool bisok(int row,int col)//函数作用:判断黑皇后,同一列,同一对角线不能相等
{
for(int i=0;i<row;i++)
if(bmap[i]==col||fabs(bmap[i]-col)==row-i)
return false;
return true;
}
void B_Q(int row)//处理黑皇后 注意我先填白皇后,因为函数调用时,必须存在,所以B_Q要在W_Q前存在,详见61行
{
if(row==n){
ans++;
return;
}
for(int i=0;i<n;i++){
if(!v[row][i]) continue;
if(wmap[row]==i) continue;//判断是否被白皇后占位
if(bisok(row,i)&&!bmap[row]){
bmap[row]=i;
B_Q(row+1);
bmap[row]=0;
}
}
}
void W_Q(int row)
{
if(row==n){
B_Q(0); //白皇后完成,填黑皇后,调用B_Q所以,B_Q要在W_Q前存在
return;
}
for(int i=0;i<n;i++){
if(!v[row][i]) continue;//判断能不能放
if(wisok(row,i)&&!wmap[row]){
wmap[row]=i;
W_Q(row+1);
wmap[row]=0;
}
}
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
cin>>v[i][j];
W_Q(0);//先填白皇后
cout<<ans;
return 0;
}