Codeup 八皇后 n皇后问题

题目：

会下国际象棋的人都很清楚：皇后可以在横、竖、斜线上不限步数地吃掉其他棋子。如何将8个皇后放在棋盘上（有8 * 8个方格），使它们谁也不能被吃掉！这就是著名的八皇后问题。 对于某个满足要求的8皇后的摆放方法，定义一个皇后串a与之对应，即a=b1b2...b8，其中bi为相应摆法中第i行皇后所处的列数。已经知道8皇后问题一共有92组解（即92个不同的皇后串）。 给出一个数b，要求输出第b个串。串的比较是这样的：皇后串x置于皇后串y之前，当且仅当将x视为整数时比y小。

采用递归的方法，递归的过程中需要判断当前的序列是否合法。合法则将当前序列转化为数字，存入a数组中。

AC代码：（此代码在codeup可以ac，在牛客网无法通过任何测试用例，但是我枚举了全部92个用例，用牛客的自测系统全部通过。很迷）

//问题 D: 八皇后
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
const int maxn = 11;
//P[i]为第i行皇后所在的列号
int p[maxn], hashTable[maxn] = { false }, a[100] = { 0 };
int n, count_ = 1;//count为总合法排列数
int arrayToNum(int* p)
{
	int temp = p[1];
	for (int i = 2; i <=n; i++) {
		temp = temp * 10 + p[i];
	}
	return temp;
}
void generateP(int index)//index代表当前填入p[index]
{
	if (index == n + 1) {
		bool flag = true;
		for (int i = 1; i <= n; i++) {
			for (int j = i + 1; j <= n; j++) {
				if (abs(i - j) == abs(p[i] - p[j]))
					flag = false;
			}
		}
		if (flag) {//当前方案合法，存入数组
			a[count_++] = arrayToNum(p);
		}
		return;
	}
	for (int i = 1; i <= n; i++) {
		if (hashTable[i] == false) {
			p[index] = i;
			hashTable[i] = true;
			generateP(index + 1);
			hashTable[i] = false;
		}
	}
}
bool cmp(int a, int b)
{
	return a < b;
}
int main()
{
	n = 8;
	generateP(1);
	sort(a, a + count_, cmp);
	int query, temp;
	scanf("%d", &query);
	while (query--)
	{
		scanf("%d", &temp);
		printf("%d\n", a[temp]);
	}
	return 0;
}

采用回溯法，可以减少很大的计算量。回溯法：如果在到达递归边界之前，由于有些已经存在的事实导致，接下来的子问题已经不可能有解，不需要再进行递归，直接返回上一层。

下面的修改就是在放置第index列的皇后于x行时，检查其与之前的1到index-1列皇后是否有冲突。这样就确保了进行到index==n+1时，1-index这整个序列必然合法。

void generateP(int index)
{
	if (index == n + 1) {
		a[count_++] = arrayToNum(p);
		return;
	}
	for (int x = 1; x <=n; x++) {
		if (hashTable[x] == false) {//第x行没有皇后，尝试把第index列的皇后放在第x行
			bool flag = true;
			for (int pre = 1; pre < index; pre++) {
				if (abs(index - pre) == abs(x - p[pre])) {
					flag = false;
					break;
				}
			}
			if (flag) {//若第index列的皇后与之前1，index-1的皇后没有冲突
				hashTable[x] = true;
				p[index] = x;
				generateP1(index + 1);
				hashTable[x] = false;
			}
		}
	}
}

从耗时上也可以看出明显的差异。