hpu1218 2n皇后问题

1218: 2n皇后问题 [搜索]

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

题目描述

给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。

现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。

问总共有多少种放法？

输入

输入的第一行为一个整数n，表示棋盘的大小。

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

n小于等于8。

输出

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

样例输入

4
1111
1111
1111
1111
4
1011
1111
1111
1111

样例输出

2
0

思路：wa了经典的两次，黑白皇后要分别搜索，记录黑皇后之后，用原来的思路解白皇后；

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
char map[1010][1010];
int visit[3][1010],vis[3][1010],mapp[1010][1010];
int n,res;
void DFS(int x,int y)//搜索白皇后；
{
	if(y==n)
	{
		res++;//对于1个情况的黑皇后，搜索出此情况下的所有白皇后；
		return;
	}
	for(int i=x;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(map[i][j]=='1' && !vis[0][j] && !vis[1][j+y]  && !vis[2][y-j+n]&& !mapp[i][j])//加个黑皇后的位置判断；
			{
				vis[0][j]=vis[1][j+y]=vis[2][y-j+n]=1;
				DFS(i+1,y+1);
				vis[0][j]=vis[1][j+y]=vis[2][y-j+n]=0;
			}
		}
	}
}
void dfs(int x,int y)
{
	if(y==n)
	{
		DFS(0,0);
		memset(vis,0,sizeof(vis));
		return;
	}
	for(int i=x;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(map[i][j]=='1' && !visit[0][j] && !visit[1][j+y] && !visit[2][y-j+n])
			{       //分别对应列，副对角线，主对角线（同n皇后问题）
				visit[0][j]=visit[1][j+y]=visit[2][y-j+n]=1;
				mapp[i][j]=1;
				dfs(i+1,y+1);
				visit[0][j]=visit[1][j+y]=visit[2][y-j+n]=0;
				mapp[i][j]=0;
			}
		}
	}
}
int main()
{
	int i,j;
	while(scanf("%d",&n)!=EOF)
	{
		res=0;
		memset(visit,0,sizeof(visit));
		memset(vis,0,sizeof(vis));
		memset(map,0,sizeof(map));
		memset(mapp,0,sizeof(mapp));
		for(i=1;i<=n;i++)
		{
			getchar();
			for(j=1;j<=n;j++)
			{
				scanf("%c",&map[i][j]);
			}
		}
		dfs(0,0);
		printf("%d\n",res);
	}
	return 0;
}