洛谷 P1219 [USACO1.5] 八皇后 Checker Challenge

题目描述

一个如下的 6×66×6 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

上面的布局可以用序列 2 4 6 1 3 52 4 6 1 3 5 来描述，第 �i 个数字表示在第 �i 行的相应位置有一个棋子，如下：

行号 1 2 3 4 5 61 2 3 4 5 6

列号 2 4 6 1 3 52 4 6 1 3 5

这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出，解按字典顺序排列。
请输出前 33 个解。最后一行是解的总个数。

输入格式

一行一个正整数 �n，表示棋盘是 �×�n×n 大小的。

输出格式

前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

输入输出样例

输入 #1复制

6

输出 #1复制

2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4

说明/提示

【数据范围】
对于 100%100% 的数据，6≤�≤136≤n≤13。

题目翻译来自NOCOW。

USACO Training Section 1.5

分析：

该题主要难点是直接暴力深搜会超时，需要寻找一个方法模拟。

        我们正常思维可能是想要构建一个二维数组去暴力dfs但是会发现在实现的时候会嵌套很多for循环，不容易写而且很容易超时。解决方案呢就是把一个二维数组拆分成三个一维数组，分别是代表列的数组 lie[ ]， 代表主对角线 (从左上到右下方向) 的数组 zhu[ ]，代表副对角线  (从右上到左下) 的数组 fu[ ]。那么有同学问了，为什么是这三个呢？为什么没有代表行的数组呢？

        我们不难看出，其实有多少行多少列就有多少个皇后。我们以行为标准，每一行选定了皇后之后是不是皇后所在的那一列肯定不会再有皇后了，所以我们一整列抽象一下，把二维数组抽象成一个一维数组lie[ ]，举个例子，如果第一行里的皇后在第二列，那我们就把lie[2]标记一下，证明这一整列都不可能有皇后。这就是为什么我们可以把行抽象没了。  那对于zhu[]和fu[]呢？我们也是把二位抽象成一维，原理是一样的让一维直接代表一行，难点是如何抽象，给大家一张图，应该可以看懂：

以样例为例子：

我们这样就可把zhu[]表示出来了，同理fu[]也是一样的（没展示出来）。OK，这样我们前置工作就做完了。

        怎么写代码呢，我们的思想dfs行，dfs（x）x是目前搜到了第几行，然后再以列为准遍历，如果这一列 lie 没有被标记（没有皇后）并且主对角和副对角线也都没有皇后zhu fu，那么我们就用arr记录下来，然后把这个点所在的列lie 主对角zhu 副对角fu 都标记一下...最终，如果x>n,x遍历到了最后一行，那么就把arr全输出出来。  ps：这里会涉及到皇后所在的列和主对角副对角怎么求的问题，我们在遍历的时候是用 i 遍历列，所以lie的话自然就是lie[ i ],但是主 副对角就比较麻烦，我们可以结合我上边给大家的图找规律，发现对主对角线zhu和i 的关系是：zhu[i-x-n] 副对角线是fu[x+i-1] （x是dfs遍历到了第几行，n是有几行几列几个皇后）

OK！上代码！

代码：

#include<iostream>
using namespace std;
int n, ans;
int arr[15];
bool zhu[27],fu[27],lie[27];//分别表示主对角线，副对角线和每一列

void dfs(int x)//第x个点,同时也代表第x行
{
	if (x > n)
	{
		ans++;//记录个数
		if (ans <= 3)
		{
			for (int i = 1; i <= n; i++)
				cout << arr[i] << ' ';
			cout << endl;
		}
		return;
	}

	for (int i = 1; i <= n; i++)//第i列
	{
		if (!lie[i]&&!zhu[i-x+n]&&!fu[x+i-1])
		{
			arr[x] = i;
			lie[i] = true; zhu[i-x+n] = true;
			fu[x+i-1] = true;
			dfs(x + 1);
			lie[i] = false; zhu[i-x+n] = false;//回溯
			fu[x+i-1] = false;
		}
	}
}

int main()
{
	cin >> n;
	dfs(1);//第一个点
	cout << ans;
	return 0;
}