[USACO1.5]八皇后 Checker Challenge（C++，DFS）

题目描述

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

上面的布局可以用序列 $246135$ 来描述，第 $i$ 个数字表示在第 $i$ 行的相应位置有一个棋子，如下：

行号 $123456$

列号 $246135$

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

输入格式

一行一个正整数 $n$，表示棋盘是 $n\times n$ 大小的。

输出格式

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

样例 #1

样例输入 #1

6

样例输出 #1

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

提示

【数据范围】
对于 $100\%$ 的数据，$6\le n\le 13$。

题目翻译来自NOCOW。

USACO Training Section 1.5

解题思路：

八皇后的问题其实类似于全排列的问题，接下来先说明为什么类似，然后我们就自然知道为什么用深度优先搜索了

首先，我们回忆一下全排列的需求，就是要求n个（0~9）数字不能重复出现，求有多少种不同的顺序

那么我们稍微修改一下这道题，我们先把n的范围变为6~13，这很好理解

然后我们再修改一下“不能重复出现”这个条件，就可以得到八皇后的问题了

接下来就会发现问题其实就在于“不能重复出现”的判断

行可以通过递推的层数来避免重复

列的不重复只需要一个一维数组即可

那么对角线呢？

先来看看主对角线，我们会发现“行数-列数”可以很好地对应每一条主对角线

（图中的单元格内为行数-列数的差）

再来看看副对角线，我们又会发现“行数+列数”可以很好地对应每一条副对角线

（图中单元格内为行数+列数的和）

那么判断的问题解决了，递归主体代码如下

int n, sum = 0;//行数、方法总数
bool column_array[13] = { false };//列标记物
bool main_diagon_array[25] = { false };//主对角标记物（-12~12）
bool secondary_diagon_array[25] = { false };//副对角标记物（0~24）
int column_queen[13] = { 0 };//存储皇后位置

void depth_first_search(int line_num)//传入行数
{
	for (int i = 0; i < n; i++)//尝试每一列
	{
		if (column_array[i] || main_diagon_array[line_num - i + 12] || secondary_diagon_array[i + line_num])//检查当前列、主副对角线
		{
			continue;
		}
		column_queen[line_num] = i + 1;//注意：存储的列数需要+1
		column_array[i] = true;//标记列数
		main_diagon_array[line_num - i + 12] = true;//标记主对角线
		secondary_diagon_array[i + line_num] = true;//标记副对角线
		depth_first_search(line_num + 1);//尝试下一行
		column_array[i] = false;//取消标记
		main_diagon_array[line_num - i + 12] = false;//取消标记
		secondary_diagon_array[i + line_num] = false;//取消标记
	}
}

递归结束的条件就很好想到啦，然后根据题意判断后输出即可

if (line_num == n)//所有行均填入皇后，终止
{
	if (sum < 3)
	{
		for (int i = 0; i < n; i++)
		{
			cout << column_queen[i] << ' ';
		}
		putchar('\n');
	}
	sum++;
	return;
}

这里可能会疑问为什么这样就是按字典顺序输出的，因为我们总体上就是在尽量先填充小序号的列，得到的结果自然就是字典顺序了

完整的实现代码如下

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

int n, sum = 0;//行数、方法总数
bool column_array[13] = { false };//列标记物
bool main_diagon_array[25] = { false };//主对角标记物（-12~12）
bool secondary_diagon_array[25] = { false };//副对角标记物（0~24）
int column_queen[13] = { 0 };//存储皇后位置

void depth_first_search(int line_num)//传入行数
{
	if (line_num == n)//所有行均填入皇后，终止
	{
		if (sum < 3)
		{
			for (int i = 0; i < n; i++)
			{
				cout << column_queen[i] << ' ';
			}
			putchar('\n');
		}
		sum++;
		return;
	}
	for (int i = 0; i < n; i++)//尝试每一列
	{
		if (column_array[i] || main_diagon_array[line_num - i + 12] || secondary_diagon_array[i + line_num])//检查当前列、主副对角线
		{
			continue;
		}
		column_queen[line_num] = i + 1;//注意：存储的列数需要+1
		column_array[i] = true;//标记列数
		main_diagon_array[line_num - i + 12] = true;//标记主对角线
		secondary_diagon_array[i + line_num] = true;//标记副对角线
		depth_first_search(line_num + 1);//尝试下一行
		column_array[i] = false;//取消标记
		main_diagon_array[line_num - i + 12] = false;//取消标记
		secondary_diagon_array[i + line_num] = false;//取消标记
	}
}

int main()
{
	cin >> n;
	depth_first_search(0);
	cout << sum << endl;
	return 0;
}