2n皇后问题

蓝桥杯基础练习VIP-2n皇后问题

时间限制: 1s 内存限制: 128MB

题目描述

给定一个n*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

代码

#include <iostream>
#include <bits/stdc++.h>//'万能'头文件
using namespace std;
int n;
int a[9][9],white[9],black[9];
int cnt;
bool checkw(int x,int y){
	if(a[x][y]==0) return false;//不可放
	for(int i=1;i<x;i++){//前x-1行
		if(white[i]==y) return false;
		if(abs(x-i)==abs(y-white[i])) return false;
	}
	return true;
}
void dfsb(int row);//声明
void dfsw(int row){
	if(row==n+1){//可行，放黑皇后
		dfsb(1);
	}
	for(int i=1;i<=n;i++){//第一列到第n列
		if(checkw(row,i)){//符合条件
			white[row]=i;//记录第row行白皇后在第几列
			a[row][i]=0;//占位
			dfsw(row+1);
			a[row][i]=1;
			white[row]=0;
		}
	}
}
bool checkb(int x,int y){
	if(a[x][y]==0) return false;
	for(int i=1;i<x;i++){
		if(black[i]==y) return false;
		if(abs(x-i)==abs(y-black[i])) return false;
	}
	return true;
}
void dfsb(int row){
	if(row==n+1){//可行
		cnt++;
		return;
	}
	for(int i=1;i<=n;i++){
		if(checkb(row,i)){
			black[row]=i;
			a[row][i]=0;
			dfsb(row+1);
			a[row][i]=1;
			black[row]=0;
		}
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>a[i][j];//1为可放，0为不可放
		}
	}
	dfsw(1);
	cout<<cnt<<endl;
	return 0;
}