HDU2553 N皇后问题（dfs）[C/C++]

题目及翻译

题面

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

输入

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

输出

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

输入样例

1
8
5
0

输出样例

1
92
10

题目思路

N皇后是老题目了，学搜索应该都会碰到，现在才来做真是对不起
看其他题解用的都是暴力遍历来找当前点是否能使用
我找到了一个巧妙一些的方法
题目意思大概就是在一个n*n的地图上面摆n个棋子，每个棋子之间以米字延伸出去的地方都不能走
也就是不能在同一行，同一列，且两个斜线也不能走，用01图实例

-0-0-
0---0
--!--
0---0
-0-0-

在以上5*5的地图中，’-'的位置不能放棋子，‘0’的位置可以放棋子，’!'代表已经放置的棋子
具体的规则可以自己去查查，百度直接搜就有示意图
以下是解题思路：
我们要确定一个点(x,y)是否能放棋子，需要确定四个条件
1：第x行没有棋子
2：第y列没有棋子
3：从左上角经过 点(x,y) 到右下角的斜线没有棋子
4：从右上角经过 点(x,y) 到左下角的斜线没有棋子
1和2很容易判断，用两个vis记录一下x和y就行
关键是3和4，一开始我还想算他们的斜率来哈希，后来发现一个巧妙的方式：
3：用x-y或x-y来记录
4：用x+y来记录
这样做的原理：
无论是3还是4的斜线都与第0行有一个交点
比如对于(1,2)这个点来说
对于3的情况：
这是一条由(1,2)与(2,3)连成的线，与第0行交点是(0,1)，那么就可以认为这条线由(0,1)为起点向右下角延伸
对于4的情况：
这是一条由(1,2)与(2,1)练成的线，与第0行交点是(0,3)，那么就可以认为这条线由(0,3)为起点向左下角延伸
观察以上情况不难看出：
对于3的情况，与第0行交点的y值是y - x
对于4的情况，与第0行交点的y值是x + y
同理还能推出，如果以 与第0列的交点作为起点，则两个交点的x值分别是x - y和x+y
所以目前已经得出所有情况，用vis数组实现对四个条件的记录。
由于每一行只有一个棋子，所以我们从0开始，跑到第n行的时候就直接退出，统计其中的结果即可

注意事项

应该有多个重复问题，要先预处理再输出

AC代码

C/C++(几乎没有代码变更)

用时15MS 内存1712K 长度514B

#include<stdio.h>
int n;
int vis[4][50];
int res[15];//预处理结果
int dfs(int i){//直接从行开始算
	if(i == n) return 1;//超过n-1行就结束
	int sum = 0;
	for(int j=0;j<n;++j) {//遍历这一行每一个位置
		if(vis[1][j])continue;//如果这一列中有棋子就跳过
		if(vis[2][i+j])continue;//斜线/中有棋子跳过
		if(vis[3][i-j+n])continue;//斜线\中有棋子就跳过
		vis[1][j] = 1;//标记这一列有棋子
		vis[2][i+j] = 1;//标记这个斜线/有棋子
		vis[3][i-j+n] = 1;//标记这个斜线\有棋子
		sum += dfs(i + 1);//统计下一步的答案
		vis[1][j] = 0;//回溯
		vis[2][i+j] = 0;//回溯
		vis[3][i-j+n] = 0;//回溯
	}
	return sum;
}
int main(){
	for(n=1;n<=10;++n){//预处理，不用清空的原因是有回溯
		res[n] = dfs(0);//记录本次的结果
	}
	while(scanf("%d",&n),n){//循环输入，n == 0时退出
		printf("%d\n",res[n]);
	}
	return 0;
}

本文作者 CSDN@扶她小藜
个人主页链接 https://blog.csdn.net/weixin_44579869