HDU 2553 N皇后问题（dfs）

Problem Description

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

Input

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

Output

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

Sample Input

1
8
5
0

Sample Output

1
92
10

代码

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;

int n,tot;
int col[11];

bool check(int r,int c)
{
	for(int i=1;i<r;i++)
		if(col[i]==c||abs(col[i]-c)==abs(i-r))
			return 0;
	return 1;
}

void dfs(int r)
{
	if(r==n+1)
	{
		tot++;
		return;
	}
	for(int c=1;c<=n;c++)
		if(check(r,c))
		{
			col[r]=c;
			dfs(r+1);
		}
}

int main()
{
	int ans[11]={0};
	for(n=1;n<=10;n++)
	{
		memset(col,0,sizeof(col));
		tot=0;
		dfs(1);
		ans[n]=tot;
	}
	while(cin>>n)
	{
		if(n==0)
			return 0;
		cout<<ans[n]<<endl;
	}
}