N皇后问题（普通递归、回溯、位运算）

哈喽，大家好，我是Tony:
欢迎大家访问我的个人主页：Tony’s Blog，让我们一起站在巨人的肩膀之上！

编程题

N皇后问题是指在N*N的棋盘上摇摆N个皇后，要求任何两个皇后不同行、不同列，也不在同一条斜线上。给定一个整数n，返回n皇后的摆法有多少种？

举例

n=1，返回1
n=2或3,2皇后和3皇后问题怎么摆都不行，返回0
n=8，返回92

思路

如果在(i,j)位置（第i行第j列放置了一个皇后）接下来在哪些位置不能放置皇后？
1、整个第i行的位置都不能放置
2、整个第j列的位置都不能放置。
3如果位置（a,b）满足|a-i|==|b-j|，在同一条斜线上，也不能放置。

递归过程为（结合代码看）：
1、用一个壳套住递归process1的初始条件（0，temp,n）
2、process1中包含结束条件i==n并以列j循环（其中包含判断fun2(i,temp,j)）

普通递归代码

#include "stdafx.h"
#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;

//普通递归
bool fun2(int i1, vector<int>temp, int j1)
{
	
		int k;
	for (k = 0; k < i1; k++)
	{
		if ((j1 == temp[k] )|| (abs(k - i1) == abs(temp[k] - j1)))
		{
			return false;
		}
		
	}
	return true;
}
int process1(int i, vector<int>temp, int n)//i为行
{
	int res =0;
	
  if (i == n)
	{
		res = 1;
		return res;
		
	}
   int j;//列
   for (j = 0; j < n; j++)
   {
	   if (fun2(i, temp, j) == true)
	   {
		   temp[i] = j;
		   res+=process1(i + 1, temp, n);
		
	   }
   }

	return res;
}
int fun(int n)
{
	
	if (n <= 1)
	{
		return -1;
	}
	vector<int>temp;//temp[i]表示第i行皇后所在的列数
	temp.resize(n);
	return process1(0, temp, n);

}
int main()
{
	int i, j;
	int n = 8;
	int res = fun(n);
	cout << res;
	return 0;

}

回溯代码

#include "stdafx.h"
#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;
int n = 8;
int res = 0;
bool fun2(int i1, vector<int>temp, int j1)
{

	int k;
	for (k = 0; k < i1; k++)
	{
		if ((j1 == temp[k]) || (abs(k - i1) == abs(temp[k] - j1)))
		{
			return false;
		}

	}
	return true;
}
int process1(int i, vector<int>temp, int n)//i为行
{
	

	if (i == n)
	{
		res++;//注意此处与普通递归的区别，回溯中res是累加状态形式
	

	}
	int j;//列
	for (j = 0; j < n; j++)
	{
		if (fun2(i, temp, j) == true)
		{
			temp[i] = j;
			process1(i + 1, temp, n);

		}
	}

	return res;
}
int fun(int n)
{

	if (n <= 1)
	{
		return -1;
	}
	vector<int>temp;//temp[i]表示第i行皇后所在的列数
	temp.resize(n);
	return process1(0, temp, n);

}
int main()
{
	int i, j;
	int n = 8;
	int res = fun(n);
	cout << res;
	return 0;

}

位运算

#include "stdafx.h"
#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;

//位运算
const int N = 8; 
int upper_limit = (1 << N) - 1; 
int ans = 0;
int fun2(int row, int ld, int rd)
{
	if (row != upper_limit) {
		int pos = upper_limit & ~(row | ld | rd);
		while (pos != 0) {
			int p = pos & (-pos);

			pos = pos ^ p;
			fun2(row | p, (ld | p) << 1, (rd | p) >> 1);
		}

	}
	else
		ans++;
	return ans;
}

int _tmain(int argc, _TCHAR* argv[])
{
	
	int n = 8;
	int res = fun2(0,0,0);
	cout << res;
	return 0;
}

代码结果