HDU - 2553：N皇后问题

N皇后问题

来源：HDU

题目

在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。
你的任务是，对于给定的N，求出有多少种合法的放置方法。

输入

共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

输出

共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

输入样例

1
8
5
0

输出样例

1
92
10

参考代码

#include
#include
const int maxn=25;
int arr[maxn];
int vis[3][maxn];//用以检测当前尝试的皇后所在的列和两个对角线是否已经存在其他皇后
int n,cnt;
void search(int row)
{
	if(row==n)
		cnt++;
	else 
	for(int i=0;i<n;i++)
	{
		if(!vis[0][i] && !vis[1][row-i+n] && !vis[2][row+i]) 
		{
			vis[0][i]=vis[1][row-i+n]=vis[2][row+i]=1;
			search(row+1);
			vis[0][i]=vis[1][row-i+n]=vis[2][row+i]=0;
		}
	}
}

int main()
{
	for(int i=1;i<=10;i++)
	{
		n=i;
		cnt=0;
		memset(vis,0,sizeof(vis));
		search(0);
		arr[n]=cnt;
	}
	
	while(scanf("%d",&n)==1 && n)
		printf("%d\n",arr[n]);
	
	return 0;
}