N皇后-回溯-位运算解法

N皇后问题是比较经典的算法问题。题目意思这里不在介绍，下面介绍两种回溯解法。

解法一：

由于我们固定每个皇后占据一行，所以简化问题为从第一行的皇后开始，为每行的皇后找到合适的列放置。

所以我们要三个数组表示每一列，正斜线以及反斜线上能否放置。

#include <iostream>
using namespace std;

const int MAX = 20;  
bool a[MAX+1] ; //列上是否有皇后，为false则有皇后，不能放置。
bool b[2*MAX + 1] ;//行和列之和相等的斜线
bool c[2*MAX + 1]; //航和列之差相等的斜线

int sum = 0;  //解法总数
int n ;  //皇后的个数
int x[MAX + 1]={0}; //每个皇后放置的列数，只求解法总数的时候可以不要


void NQueen(int i)
{
	for(int j = 1;j<=n;j++)
	{
		if(a[j] && b[i+j] && c[i-j+n]) //找到可以放置皇后的列
		{
			x[i] = j;
			a[j] = false;    //一个皇后放好，相应的列不能放置，设为false；
			b[i+j] = false;
			c[i-j+n] = false;  //i-j有可能为负，都要加上n
			if(i == n)     //表示n个皇后都放好了，这是一种成功的放置方法
				sum++;
			else
				NQueen(i+1);     //用递归调用，自然回溯
 
			a[j] = true;
			b[i+j] = true;
			c[i-j+n] = true;
		}
		
	}
}


int main()
{
	cin >> n;
	
	for(int i = 0;i<=n;i++)
		a[i] = true;
	for(int i = 0;i<=2*n;i++)
	{
		b[i] = true;
		c[i] = true;
	}

	NQueen(1);   //从第一行的皇后开始

	cout << sum << endl;

	return 0;
}

在网上找解法的时候，看到有大神说用位运算代替上面a，b，c三个数组表示列的可用性，算法的效率会提高很多。

算法思想：使用bit数组来代替以前由int或者bool数组来存储当前格子被占用或者说可用信息，从这可以看出N个皇后对应需要N位表示。
以前我们需要在一个N*N正方形的网格中挪动皇后来进行试探回溯，每走一步都要观察和记录一个格子前后左右对角线上格子的信息；采用bit位进行信息存储的话，就可以只在一行格子也就是（1行×N列）个格子中进行试探回溯即可，对角线上的限制被化归为列上的限制。

代码如下：

<pre name="code" class="cpp">#include <stdio.h>
#include <stdlib.h>
#include <time.h>

long sum = 0, upperlim = 1;   // sum用来记录排列总数，upperlim用来作bit mark。
                              // 有多少个皇后，就有多少bit被置1。（从低位到高位）

// 放置顺序是从右边列开始的。
void test(long row, long ld, long rd) / row记录了列的放置信息。bit为1时代表已经放置了皇后。
// ld记录了上一个皇后的左边1列不能放置皇后。
//rd记录了左边1列不能放置皇后。
// 如：上一次放的是第6列: 则
// row  ****  ****  **1*  ****
// ld   ****  ****  *1**  *****
// rd   ****  ****  ***1  ****
{

	if (row != upperlim)
	{
		long pos = upperlim & ~(row | ld | rd);  // 将row，ld，rd同时为0的bit提出来。
		while (pos)
		{
			long p = pos & -pos; // 将pos除最低的为1的bit外，所有的位清零后赋值给p
                           // pos不改变。（取得可以放皇后的最右边的列）

			pos -= p;  // 将pos的最低的为1的bit清零。（次右边的列，为循环做准备）
			test(row + p, (ld + p) << 1, (rd + p) >> 1);  // row + p将当前列置1，表示放上了皇后
                                             // (ld + p) << 1。设置左边列不能放。
		}
	}
	else         // row的每一位为1，即所有皇后都放好了。
		sum++;
}

int main(int argc, char *argv[])
{
	int n;
	scanf("%d",&n);
	time_t tm;

	tm = time(0);
	if ((n < 1) || (n > 32))
	{
		printf(" 只能计算1-32之间\n");
		exit(-1);
	}
	upperlim = (upperlim << n) - 1;

	test(0, 0, 0);
	printf("共有%ld种排列, 计算时间%d毫秒 \n", sum, (int) (time(0) - tm)*1000);
	return 0;
}

原帖地址：点击打开链接