2n皇后问题

给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。

输入格式:

输入的第一行为一个整数n，表示棋盘的大小。 　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。

输出格式:

输出一个整数，表示总共有多少种放法。

输入样例1:

4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1

输出样例1:

2

输入样例2:

4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1

输出样例2:

0

#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
using namespace std;
int chess[9][9];//棋盘
int wqueen[9];//白皇后放置情况
int bqueen[9];//黑皇后放置情况
int n,count;
int check(int array[],int row,int col){
	if(!chess[row][col])
		return 0;
	for(int i=1;i<row;i++){
		if(array[i]==col||abs(array[i]-col)==row-i)
			return 0;
	}
	return 1;
}
void wdfs(int row){//白皇后搜索
	if(row==n+1){//白皇后搜到n+1行，可行放置数增加
		count++;
		return;
	}
	for(int i=1;i<=n;i++){
		if(i==bqueen[row])//此位置已放置黑皇后
			continue;
		if(check(wqueen,row,i)){
			wqueen[row]=i;
			wdfs(row+1);
		}
	}
}
void bdfs(int row){//先黑皇后搜索
	if(row==n+1){//黑皇后搜索结束，搜索白皇后
		wdfs(1);
		return;
	}
	for(int i=1;i<=n;i++){
		if(check(bqueen,row,i)){
			bqueen[row]=i;
			bdfs(row+1);
		}
	}
}
int main(){
	cin>>n;
	count=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&chess[i][j]);
		}
	}
	bdfs(1);
	cout<<count<<endl;
	return 0;
}