UVa 639 Don't Get Rooked

此题跟八个皇后有些相似......只不过加了“墙”......要注意DFS参数设定

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=5;
int n,ans=0;
int map[MAXN][MAXN];
int vis[MAXN][MAXN];
bool ok(int x, int y)
{
	if(map[x][y] || vis[x][y])
		return false;
	for(int i=x-1; i>=0 && !map[i][y]; i--)	//判断列是否合法
		if(vis[i][y])
			return false;
	for(int i= y-1; i>=0 && !map[x][i]; i--)	//判断行是否合法
		if( vis[x][i])
			return false;
	return true;
}
void dfs(int x,int y, int num)
{
	if(x == n)
	{
		if(num >ans)
			ans=num;
		return;
	}
	if(ok(x, y))
	{
		vis[x][y] =1;
		if(y >= n-1)
			dfs(x+1, 0, num+1);
		else
			dfs(x, y+1, num+1);
		vis[x][y]=0;
	}//在这一定不能有else，不然无法遍历矩阵全部数据
	if(y >= n-1)
		dfs(x+1, 0, num);
	else
		dfs(x, y+1, num);
}
int main()
{
	char s[MAXN];
	while(scanf("%d", &n) != EOF && n)
	{
		memset(map, 0, sizeof(map));
		memset(vis, 0, sizeof(vis));
		for(int i=0; i<n; i++)
		{
			scanf("%s", s);
			for(int j=0; j<n; j++)
				if(s[j] == 'X')
					map[i][j]=1;
		}
		ans=0;
		dfs(0,0,0);
		printf("%d\n", ans);
	}
	return 0;
}