试题 基础练习 2n皇后问题

问题描述

     给定一个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;
}