八皇后问题 (源于 刘汝佳的《算法竞赛入门经典(第2版)》)
在棋盘上放置8个皇后,使得它们互不攻击,此时每个皇后的攻击范围为同行同列和同对角线,要求找出所有解。(这里只输出解的个数)
【分析】
棋盘为8行8列,可想而知,放置8个皇后,想让8个皇后互不攻击,那么它们至少每一个都要放置在不同行不同列中,即恰好每行每列各放置一个皇后。如果用C[x]表示第x行皇后的列编号,则问题变成了全排列生成问题。就是列编号0~7进行全排列,而每个列编号则依次储存在C[x]中,那么第x行皇后就有对应的列编号。在递归枚举的过程中,当我们发现该解不合法的时候,就停止对该解继续递归的操作,而且返回上一级的递归调用继续求解。(回溯)
注:当把问题分成若干步骤并递归求解时,如果当前步骤没有合法选择,则函数将返回上一级递归调用,这种现象称为回溯。
【程序】
用C++语言编写程序,代码如下:
(两种求解八皇后问题的方法)
#include<iostream>
using namespace std;
int tot;
const int n = 8;
int C[n];
void search1(int cur) {
if (cur == n) tot++; //递归到边界。只要走到这里,所有皇后必然不冲突
else {
for (int i = 0; i < n; i++) {
int ok = 1;
C[cur] = i; //尝试把第cur行的皇后放在第 i 列
for (int j = 0; j < cur; j++) //检查是否和前面的皇后冲突
if (C[j] == C[cur] || cur - C[cur] == j - C[j] || cur + C[cur] == j + C[j]) {
ok = 0;
break;
}
if (ok)
search1(cur + 1); //如果合法,则继续递归
}
}
}
int vis[3][100];
void search2(int cur) {
if (cur == n) tot++;
else for (int i = 0; i < n; i++) {
//利用二维数组直接判断
//vis数组表示已经放置的皇后占据了哪些列、主对角线和副对角线
if (!vis[0][i] && !vis[1][cur + i] && !vis[2][cur - i + n]) {
C[cur] = i; //如果不用 打印解,整个C数组都可以省略
vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 1; //修改全局变量
search2(cur + 1);
vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 0; //恢复原状
}
}
}
int main() {
tot = 0;
search1(0);
cout << "search1 : " << endl;
cout << tot << endl;
tot = 0;
search2(0);
cout << "search2 : " << endl;
cout << tot << endl;
return 0;
}