问题描述
给定一个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
样例输出
2
样例输入
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
0
代码:
#include<iostream>
using namespace std;
#include<math.h>
int a[8][8]; //棋盘
int posB[8]; //每一行黑皇后的位置
int posW[8]; //每一行白皇后的位置
int count=0;
bool checkW(int row)
{
if(a[row][posW[row]]==0)return false; //原棋盘不能放皇后
if(posB[row]==posW[row])return false; //先放了黑皇后
for(int i=0;i<row;i++)
{
if(posW[i]==posW[row])return false; //同一列
//if(row==posW[row]&&i==posW[i])return false; //正对角线
//if(row+posW[row]==n-1&&i+posW[i]==n-1)return false; //反对角线
if(fabs(i-row)==fabs(posW[i]-posW[row]))return false;
}
return true;
}
void dfsW(int i,int n)
{
if(i==n)
{
count++;
return ;
}
for(int j=0;j<n;j++)
{
posW[i]=j;
if(checkW(i))
{
dfsW(i+1,n);
}
}
}
bool checkB(int row)
{
if(a[row][posB[row]]==0)return false; //原棋盘不能放皇后
for(int i=0;i<row;i++)
{
if(posB[i]==posB[row])return false; //同一列
//if(row==posB[row]&&i==posB[i])return false; //正对角线
//if(row+posB[row]==n-1&&i+posB[i]==n-1)return false; //反对角线
if(fabs(i-row)==fabs(posB[i]-posB[row]))return false; //同一连线
}
return true;
}
void dfsB(int i,int n)
{
if(i==n)
{
dfsW(0,n);
return ;
}
for(int j=0;j<n;j++)
{
posB[i]=j;
if(checkB(i))
{
dfsB(i+1,n);
}
}
}
int main()
{
int n;
cin>>n;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
cin>>a[i][j];
}
}
dfsB(0,n);
cout<<count;
return 0;
}