C语言 2n皇后问题

问题描述
　　给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。
　　
输入格式
　　输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。
　　
输出格式
　　输出一个整数，表示总共有多少种放法。

#include<stdio.h>
#define MaxSize 10
//注意数组存储下标代表每一行，数组元素代表皇后在每一行的位置
int hei[MaxSize];//黑皇后
int bai[MaxSize];//白皇后
int chessboard[MaxSize][MaxSize];//1:能放  0:不能放
int count = 0;//记录共有多少种放皇后的位置
int Check(int *str, int line)//该函数判断皇后是否能够放置在str[line]这个位置
{
	int i = 1;
	for (;i < line;i++)
	{
		int temp = str[i] - str[line];
		if (temp == 0 || temp == i - line || temp == line - i)
			//temp == 0，表示皇后在同一列；temp == i - line 和 temp == line - i代表皇后在正反对角线
		{
			return 0;
		}
	}
	return 1;
}

void Whilte(int n, int line)
{
	int i; 
	if (line == n + 1)
	{
		count++;
	}
	else
	{
		for (i = 1;i <= n;i++)//i空值第line行的每一列
		{
			if (chessboard[line][i] == 1&&hei[line]!=i)//判断该位置是否可以放皇后
			{
				bai[line] = i;//记录第line行皇后的位置
				if (Check(bai, line))//进行判断，满足条件就递归下一行
				{
					Whilte(n, line + 1);
				}
			}
		}
	}
}

void Black(int n, int line)//该函数放黑皇后
{
	int i;
	if (line == n+1)//如果行line等于n+1;说明该次递归找到一次所有黑皇后满足条件的位置，就去放白皇后
	{
		return Whilte(n, 1);
	}
	else
	{
		for (i = 1;i <= n;i++)//i控制第line行的每一列
		{
			if (chessboard[line][i] == 1)//判断该位置是否可以放黑皇后
			{
				hei[line] = i;//记录第line行黑皇后的位置
				if (Check(hei, line))//进行判断，满足条件就递归下一行
					//注意该递归是放在循环内部的，所以当第i列不满足条件时，就会退回i-1列继续判断，即回溯
				{
					Black(n, line + 1);
				}
			}
		}
	}
}

int main()
{
	int n,i,j;
	printf("请输入棋盘大小n:\n");
	scanf("%d", &n);
	printf("请输入棋盘元素；1：该位置可以放皇后；0：该位置不可以放皇后\n");
	for (i = 1;i <= n;i++)
	{
		for (j = 1;j <= n;j++)
		{
			scanf("%d", &chessboard[i][j]);
		}
	}
	for (i = 1;i <= n;i++)
	{
		for (j = 1;j <= n;j++)
		{
				printf("%d", chessboard[i][j]);
		}
		printf("\n");
	}
	Black(n, 1);
	printf("皇后共有%d种放法\n", count);
	return 0;
}