Bootstrap

【回溯算法】N皇后问题

【问题描述】

在n×n格的棋盘上放置彼此不受攻击的n个皇后。

按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。n皇后问题等价于在n×n格的棋盘上放置n个皇后,任何两个皇后不放在同一行或同一列或同一斜线上。

编程要求:找出一个n×n格的棋盘上放置n个皇后并使其不能互相攻击的所有方案。

 

【算法分析】


由于棋盘的每列只有一个皇后,所以可以用一维向量X( x1, x2, …, xn),其中xi∈{1, 2, …, n},表示第i列皇后所在的行x[i],即解空间的每个结点都有n个儿子,因此解空间的大小为n的n次方,这是一棵子集树。【算法分析】

【代码部分】

//n皇后问题回溯算法的数据结构
#define NUM 20
int n;			//棋盘的大小
int x[NUM];		//解向量
int sum;		//当前已经找到的可行方案数
//N皇后问题回溯算法的实现
//形参t是回溯的深度,从1开始
void Backtrack(int t) 
{ 
  int i;
  //到达叶子结点,获得一个可行方案。累计总数,并输出该方案
  if (t>n) 
  {
    sum++;		//是全局变量
    for (i=1; i<=n; i++) 
      printf(" %d", x[i]);
    printf("\n");
  }
  else
    for (i=1; i<=n; i++) 
    {
      x[t] = i;
      if (Place(t)) Backtrack(t+1);
    }
}
//检查当前皇后位置的约束函数
//形参t是回溯的深度
inline bool Place(int t) 
{ 
  int i; 
  for (i=1; i<t; i++) 
    if ((abs(t-i) == abs(x[i]-x[t])) || (x[i] == x[t])) 
      return false; 
  return true; 
}

由于每一列只放置一个皇后,所以不用判断合法性。

对于每一行,假设已经放置到t列,只要判断 ,i=1, 2, …, t-1互不相同即可。

对于对角线的判断,可以看成是斜率为±1的两条直线,经过两点(i,x[i])和(t,x[t]):

//算法的完整实现
#include <iostream> 
#include <cmath>
using namespace std;

#define NUM 20
int n;
int x[NUM]; 
int sum; 

inline bool Place(int t) 
{ 
	int i; 
	for (i=1; i<t; i++) 
		if ((abs(t-i) == abs(x[i]-x[t])) || (x[i] == x[t])) 
			return false; 
	return true; 
} 

void Backtrack(int t) 
{ 
	int i;
	if (t>n) 
	{
		sum++;
		for (i=1; i<=n; i++) 
			printf(" %d", x[i]);
		printf("\n");
	}
	else
		for (i=1; i<=n; i++) 
		{
			x[t] = i;
			if (Place(t)) Backtrack(t+1);
		}
} 

int main() 
{ 
	while (cin>>n)
	{
		sum = 0;
		Backtrack(1);
		printf("Total= %d\n\n", sum);
	}
	return 0; 
} 

 

;