基础练习 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

思路：
先放置一种皇后 标记已放置皇后位 根据n*n矩阵放置n个皇后又要满足条件 所以每行必须有皇后；
放置完一种皇后再开始放另一种皇后， 不能重复放置。

可以通过 bj[][x-y+n]加 n 保证 x-y+n 为正数 防止bj[]溢出的来标记右对角 bj[][x+y]来标记左对角
bj[][y]标记列

代码：

#include <stdio.h>
int a[10][10];
int bj[3][20]={0};
int bj1[3][20]={0};
int n;
int sum=0;
void f2(int x){//放置黑后 
	int y;
	if(x==n)
		sum++;
	else{
		for(y=0;y<n;y++)
			if(a[x][y]==1&&bj1[0][y]==0&&bj1[1][x+y]==0&&bj1[2][x-y+n]==0){
				a[x][y]=3;
				bj1[0][y]=bj1[1][x+y]=bj1[2][x-y+n]=1;
				f2(x+1);
				bj1[0][y]=bj1[1][x+y]=bj1[2][x-y+n]=0;
				a[x][y]=1;
			}
	}
}
void f(int x){
	int y;
	if(x==n)
		f2(0);//放置黑后
	else{
		for(y=0;y<n;y++)
		if(a[x][y]==1&&bj[0][y]==0&&bj[1][x+y]==0&&bj[2][x-y+n]==0){//左右对角 列 和当前坐标都未被标记可放 
				a[x][y]=2;//标记白后的位置，防止黑后放重 
				bj[0][y]=bj[1][x+y]=bj[2][x-y+n]=1;//标记左右对角 列 
				f(x+1);
				bj[0][y]=bj[1][x+y]=bj[2][x-y+n]=0;//清除标记 
				a[x][y]=1;
			}
	} 
}
 
int main(){
	int i,j;
	scanf("%d",&n);
	for(i=0;i<n;i++)
	for(j=0;j<n;j++)
		scanf("%d",&a[i][j]);
	f(0);
	printf("%ld\n",sum); 
	return 0;
}