2n皇后

问题描述

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

   解题思路：

   深搜黑皇后的可能分布，达到n时候深搜白皇后。

   

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int ans;
bool row[10],col[10];
int maze[10][10];
int n;
bool z[20],f[20];
int c[20];
void dd(int cur)//这一段对白皇后的搜索为白书中对八皇后问题求解的代码
{
	int i,j;
	if(cur==n)
	ans++;
	else
	for(i=0;i<n;i++)
	{
		int ok=1;
		if(maze[cur][i]==0)
		continue;
		c[cur]=i;
		for(int j=0;j<cur;j++)
		{
			if(c[cur]==c[j]||cur-c[cur]==j-c[j]||cur+c[cur]==j+c[j])
			{
				ok=0;
				break;
			}
		}
		if(ok)
		dd(cur+1);
	}
}
void dfs(int count)
{
	if(count==n)
	{
		dd(0);
		return ;
	}
	for(int j=0;j<n;j++)
	{
		if(!row[count]&&!col[j]&&maze[count][j]==1&&!z[count+j]&&!f[n-1-j+count])
		{
			z[count+j]=true;
			f[n-1-j+count]=true;
			row[count]=true;
			col[j]=true;
			maze[count][j]=0;
			dfs(count+1);
			z[count+j]=false;
			f[n-1-j+count]=false;
			row[count]=false;
			col[j]=false;
			maze[count][j]=1;
		}
	}
	return ;
}
int main()
{
	cin>>n;	
	for(int i=0;i<n;i++)
	for(int j=0;j<n;j++)
	cin>>maze[i][j];
	memset(row,0,sizeof(row));
	memset(col,0,sizeof(col));
	memset(z,0,sizeof(z));
	memset(f,0,sizeof(f));
	memset(c,-1,sizeof(c));
	ans=0;
	dfs(0);
	cout<<ans<<endl;
	
}