回溯算法 --- 例题4.N皇后问题
一.问题描述
N*N棋盘上放置N个皇后使得每个皇后互不受攻击. 即任二皇后不能位于同行同列和同一斜线上.
如四皇后问题的两个解:
二.解题思路
将棋盘从左至右,从上到下编号为1,...,n,皇后编号为1,...,n.
设解为(x1, ..., xn) , xi为皇后i的列号,且xi位于第i行.
解空间:E={ (x1,..., xn) | xi∈Si, i=1,...,4}, Si={1, ..., 4},1 <= i <= n
解空间为排列树? 是的!N个皇后在N*N的棋盘中一共有C(N^2, N)种放置情况,我们需要找到其中满足条件的排列情况.
其约束集合D为:
- xi ≠ xj ,保证皇后i, j不在同一行中
- xi - i ≠ xj - j ,保证皇后i, j不在同一斜线上
- xi + i ≠ xj +j ,保证皇后i, j不在同一斜线上
后两条总结一下就是 abs(i-j) ≠ abs(x[i] - x[j])
用回溯法解N后问题时,用完全n叉树表示解空间.可行性约束Place减去不满足行,列和斜线约束的子树.
具体代码如下:
// N皇后问题回溯算法
#include
using namespace std;
class Queen
{
friend int NQueen(int) ;
private:
bool Place(int k);
void Print();
void Backtrack(int t);
int n; //皇后个数
int *x; //当前解
long sum; //当前已经找到的可行方案数
};
void Queen::Print()
{
cout<<"放置方案为: ";
for(int i=1; i<=n; i++) cout< n)
{
sum++;
Print();
}
else
{
for(int i=1; i<=n; i++) //遍历第t行每一列
{
x[t] = i;
if(Place(t)) //如果第t行能够放置
{
Backtrack(t+1);
}
}
}
}
int NQueen(int n)
{
Queen X;
//初始化X
X.n = n;
X.sum = 0;
int *p = new int[n+1];
for(int i=0; i<=n; i++) p[i] = 0;
X.x = p;
X.Backtrack(1);
X.Print();
delete [] p;
return X.sum;
}
int main()
{
cout<<"请输入皇后个数: ";
int n;
while(cin>>n && n)
{
int ans = NQueen(n);
cout<<"放置方案数: "<
运行结果如下:
参考毕方明老师《算法设计与分析》课件.
欢迎大家访问个人博客网站---乔治的编程小屋,一起加油!