洛谷P1219N皇后问题

题目描述

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

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

行号 1 2 3 4 5 6     

列号 2 4 6 1 3 5     

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

输入格式

一行一个正整数 n，表示棋盘是 n×n 大小的。（n <= 13）

输出格式

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

解题思路

        首先从题目意思，我们不难得知棋盘上的所有皇后必须满足【两两不在同一行，不在同一列，也不在同一斜边上】的条件，那么题目要求就可以转换成求出所有满足条件的解，这样的题目在数据量不是很大的情况下可以采用深搜的方法。我们在深搜时需要做剪枝优化以及合法判断。具体怎么优化怎么判断呢，最好想到的情况就是用数组去记录，这样当然可以，但是这次要分享的是一个状态压缩的技巧和思路。

        我们如果用一个数组来记录当前行，当前列，当前正斜边， 以及当前反斜边是否被用过的话是需要四个vis数组来存储结果的，但是我们想到，这四个从、vis数组都是只有两个状态（0和1），由此联想到，我们计算机中二进制数也是只由0和1组成，那么我们是不是可以把这四个vis数组换成四个整形来表示呢？

        让我们先来看一下题目的数据范围（n <= 13），考虑斜边，在一个n * n的正方形中，一定有（2 * n - 1）条正斜边以及（2 * n - 1） 条反斜边（如图所示），那么可以知道最多只有25条斜边，每个int类型有32位，所以是可以表示这25条斜边的！

然而现在的问题就是如何表示这每一条正斜边以及反斜边呢？找规律的话不难发现斜边存在以下的规律：

如图所示，对于每一个点（i， j）来说，可以用（i + j - 1）来表示他所在的正斜边，用(i - j + n)来表示他所在的反斜边，这样我们就可以准确的描述每一行，每一列以及每一条斜边了o(*￣▽￣*)ブ 

        接下来又是一个难题 ，我们知道了正方形行列和斜边的规律以及表示方法，上面我们提到了状态压缩，就是把一个数组压缩成一个整形来表示状态，这个该如何表示呢？

        我们知道，int整形有32位，每一位都是0或1，那么也就意味着我们可以把它看成一个32位的vis数组，现在的问题就转化成了：【怎么精确的控制每一位的0和1】，这个就涉及位运算的相关知识了，对于一个整型t， (-t & t)得到的第一位不为0的数一定是t这个数字从右往左数第一个1的位置：

 这样我们就可以得到数字t从右往左数最低位的1，这样有什么用呢，我们用vis数组判断的话也是1的时候表示用过，0表示没用过，那么我们整形也可以用类似的办法表示，但是又有一个问题，我们用数组表示可以直接用vis[i]来表示vis数组中下标为i的值，那用整形的话我们怎么快速得到第i位的数位是0还是1呢？其实也很简单，就是把1左移i位，这个位置上的1就是第i位的值（因为我们的行列从1开始，所以i也是从1开始的）。这个地方我们可以用一个ind数组来表示，如图所示：

int ind[MAXN];//关联每一位的"1"及其权值 ind[i]表示（1 << i）[也可以理解成第(i + 1)个"1"所在的位置]的权值 

//把每一位的"1"和其权值关联起来，用的时候直接ind[i & -i]即可得到这个"1"所在的权值 
	for (int i = 0; i < 2 * n; i++) ind[1 << i] = i;

 这样ind数组的下标就是第i位上面的1，而ind[i]则代表了这位1的权值（通俗点来讲就可以看成vis数组的下标），这样我们只需要计算含有n个1的掩码即可：

#define MASK(n) ((1 << ((n) + 1)) - 2)

这里我们用一个宏来计算掩码，比如n = 3的时候，（1 << (3 + 1)）- 2的话就是1110,这样从第1位开始就是3个连续的1。

        好了， 上面就是状态压缩的全过程， 现在就是dfs的过程了，话不多说直接看代码

int dfs(int i, int t1, int t2, int t3) {
	if (i > n) {
		if (cnt) print_one_result(n);
		return 1;
	}
	//用一个ans记录深搜全部过程的结果 
	int ans = 0;
	//把所有能用的列全部枚举出来 
	for (int k = t1; k; k -= (-k & k)) {
		//j就是当前可用的列的下标 
		int j = ind[-k & k];
		//判断当前列所在的正反两条斜边能否使用 
		if ((t2 & (1 << (i + j - 1))) && (t3 & (1 << (i - j + n)))) {
			//如果都能使用，则在(i, j)位置放置一个皇后 
			arr[i] = j;
			//继续深搜，在深搜过程中分别把t1, t2, t3中当前列，正反斜边标记成0（表示用过） 
			ans += dfs(i + 1, t1 ^ (1 << j),
					t2 ^ (1 << (i + j - 1)), t3 ^ (1 << (i - j + n))
					);
		}
	}
	//返回结果 
	return ans;
}

这里呢就是dfs函数的实现过程，递归向下搜索的时候用到了异或运算（相同得0不同得1）来去掉当前已经用过的状态，大家手动模拟一下应该就明白啦。

 代码实现

接下来就是全部的代码实现啦！

#include<iostream>
#include<bits/stdc++.h>
//掩码信息，用于初始化每个行，列 和斜边的状态 
#define MASK(n) ((1 << ((n) + 1)) - 2)
using namespace std;

//由于题目范围最大掩码不超过2的26次方，所以不超过1e8 
const int MAXN = 1e8;
int n, cnt = 3;//n用于输入棋盘大小，cnt用来控制输出前三行 
int ind[MAXN];//关联每一位的"1"及其权值 ind[i]表示（1 << i）[也可以理解成第(i + 1)个"1"所在的位置]的权值 
int arr[20];

//输出结果的函数（非常好理解(*^_^*)） 
void print_one_result(int n) {
	for (int i = 1; i <= n; i++) {
		if (i > 1) cout << " ";
		cout << arr[i];
	}
	cout << endl;
	//每次输出完输出的次数就要减一此 
	cnt -= 1;
	return ;
}

/*开始深搜的过程：
	参数解释：
	(i):当前所在的行数
	(t1):当前列的状态掩码（0表示不能使用（已经使用过），1表示可以使用（还未被使用））
	(t2):正斜边上的状态掩码，状态表示如上
	(t3):反斜边上的状态掩码，状态标识如上
*/ 
int dfs(int i, int t1, int t2, int t3) {
	if (i > n) {
		if (cnt) print_one_result(n);
		return 1;
	}
	//用一个ans记录深搜全部过程的结果 
	int ans = 0;
	//把所有能用的列全部枚举出来 
	for (int k = t1; k; k -= (-k & k)) {
		//j就是当前可用的列的下标 
		int j = ind[-k & k];
		//判断当前列所在的正反两条斜边能否使用 
		if ((t2 & (1 << (i + j - 1))) && (t3 & (1 << (i - j + n)))) {
			//如果都能使用，则在(i, j)位置放置一个皇后 
			arr[i] = j;
			//继续深搜，在深搜过程中分别把t1, t2, t3中当前列，正反斜边标记成0（表示用过） 
			ans += dfs(i + 1, t1 ^ (1 << j),
					t2 ^ (1 << (i + j - 1)), t3 ^ (1 << (i - j + n))
					);
		}
	}
	//返回结果 
	return ans;
}

int main() {
	cin >> n; cnt = 3;
	//把每一位的"1"和其权值关联起来，用的时候直接ind[i & -i]即可得到这个"1"所在的权值 
	for (int i = 0; i < 2 * n; i++) ind[1 << i] = i;
	int ans = dfs(1, MASK(n), MASK(2 * n - 1), MASK(2 * n - 1));
	cout << ans << endl;
	return 0;
}

总结

        其实这个八皇后问题本身并不难理解，这里主要是想分享一下位运算来进行状态压缩的神奇之处，也帮自己加深一下印象，本身位运算又比数组取值要快一些，在dfs过程中需要大量计算取值，这样我们巧妙地运用位运算就可以既减少空间小耗又减少时间消耗。如果觉得有点帮助的话帮忙点个小赞，让更多人可以看到吧~