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

【题意】：就是非常经典的n皇后问题。记得去年暑假学回溯算法的时候，老师就讲到过这个经典的例子，当时第一次听觉得好牛逼，好复杂，其实现在想想，就是一个非常常规的dfs了。一个学期来，确实可以学到很多东西啊！

本题思路没什么问题，每次搜一行，每一行都有n列的选择，所以肯定有一个for循环枚举所有可能的选择。如果该列可以放皇后，那么我就尝试去放，然后递归进去搜下一个皇后的摆放位置，搜完后把这个位置的标记去掉，尝试其他列的情况，这就是回溯。剪枝语句也比较好写，记得当时刚开始学的时候，只能按照老师的那个套路写，现在完全不用，直接可以按照自己的理解把剪枝语句写完。就是45°不能放，同一列不能放，用循环扫一遍即可！

另外，本题因为数据量就是n = 10。所以我们可以借助dp的思想，将搜索过的结果存在数表dp中，这样如果输入样例有重复的测试数据，那我们也可以直接查表获得，大大提升了速度！如果不打表，闷着头搜，就会超时

AC代码：

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

const int maxn = 11;
int res = 0;
int dp[maxn];
int tag[maxn][maxn];

bool jianzhi(int k, int i, int n)
{
	for(int j = 1;j < k;j++)
	{
		if(tag[j][i] == 1)
			return false;
	}
	for(int j = 1;j < k && j < i;j++)
	{
		if(tag[k - j][i - j] == 1)
			return false;
	}
	for(int j = 1;k - j > 0 && i + j <= n;j++)
	{
		if(tag[k - j][i + j] == 1)
			return false;
	}
	return true;
}

void dfs(int k, int n)
{
	if(k == n + 1)
	{
		res++;
		return ;
	}
	for(int i = 1;i <= n;i++)
	{
		if(jianzhi(k, i, n))
		{
			tag[k][i] = 1;
			dfs(k + 1, n);
			tag[k][i] = 0;
		}
	}
}

int main()
{
	for(int i = 1;i <= 10;i++)
	{
		memset(tag, 0, sizeof(0));
		res = 0;
		dfs(1, i);
		dp[i] = res;
	}
	int m;
	while(scanf("%d", &m) && m != 0)
	{
		cout << dp[m] << endl;
	}
	return 0;
}