题目描述
八皇后问题是一个经典的排列问题,最早由国际象棋棋手马克斯·贝瑟尔(Max Bezzel)于1848年提出。问题的目标是在一个8×8的棋盘上放置八个皇后,使得它们互相之间不能互相攻击,即任意两个皇后不能位于同一行、同一列或同一对角线上。
思考
解决八皇后问题可以使用回溯算法或者递归算法。基本思想是逐行放置皇后,每一行只能放置一个皇后,并且保证当前行放置的皇后不与之前行放置的皇后相互攻击。如果在某一行无法找到合适的位置放置皇后,则回溯到上一行重新选择位置。
方法
1 回溯法
(1)确定函数参数和返回值
void solve_queen(int StartIndex,int n,int* arr,vector<vector<int>>& Ret)
(2)确定终止条件
if(!Check(n,arr)){
return;
}
//剪枝
if(StartIndex==n && Check(n,arr)){
arr==>Ret;
return;
}
(3)确定单层搜素逻辑
for(int i=StartIndex*n;i<StartIndex*n+n;++i){
arr[i]=true;
solve_queen(StartIndex+1,n,arr,Ret);
arr[i]=false;
}
代码实现
1 回溯法
#include<iostream>
#include<graphics.h>
#include<easyx.h>
#include<vector>
#include <windows.h>//sleep()函数
//初始化棋盘
void initialChessboard(int n, bool*& arr) {
arr = (bool*)malloc(sizeof(bool) * (n * n));
for (int i = 0; i < (n * n); ++i) {
arr[i] = false;
}
}
//展示棋盘
void ShowChessboard(int n,bool* arr) {
for (int i = 1; i <= n + 1; ++i) {
line(50, i * 50, 50 * (n + 1), i * 50);
line(i * 50, 50, i * 50, 50 * (n + 1));
}
for (int i = 0; i < n; ++i) {
for (int j = 0; j < n; ++j) {
if (arr[i * n + j]) {
setfillcolor(GREEN);
fillcircle((j + 1) * 50 + 25, (i + 1) * 50 + 25, 25);
setfillcolor(WHITE);
}
}
}
}
//标记棋子
void Sign(int n, int num) {
int line = num / n;
int row = num % n;
setfillcolor(RED);
fillcircle((row + 1) * 50 + 25, (line + 1) * 50 + 25, 25);
setfillcolor(WHITE);
}
//检查是否合法
bool Check(int n, bool* arr) {
//逐行检测
for (int i = 0; i < n; ++i) {
int flag = -1;//标志在一行中是否存在过棋子
for (int j = 0; j < n; ++j) {
if (arr[i * n + j]) {
if (flag!=-1) {
Sign(n, flag);
Sign(n, i * n + j);
return false;
}
flag = i*n+j;
}
}
}
//逐列检测
for (int i = 0; i < n; ++i) {
int flag = -1;//标志在一列中是否存在过棋子
for (int j = 0; j < n; ++j) {
if (arr[i + j * n]) {
if (flag!=-1) {
Sign(n, flag);
Sign(n, i + j * n);
return false;
}
flag = i + j * n;
}
}
}
//逐对角线检查
//左上到右下
for (int i = 0; i < n - 1; ++i) {
int flag = -1;
int j = i;
while (j%n || j==0) {
if (arr[j]) {
if (flag!=-1) {
Sign(n, flag);
Sign(n, j);
return false;
}
flag = j;
}
j += (n + 1);
}
}
for (int i = n; i < n*(n-1); i=i+n) {
int flag = -1;
int j = i;
while (j < n* n) {
if (arr[j]) {
if (flag!=-1) {
Sign(n, flag);
Sign(n, j);
return false;
}
flag = j;
}
j += (n + 1);
}
}
//右上到左下
for (int i = n-1; i >0; --i) {
int flag = -1;
int j = i;
while ((j+1)%n || j==n-1) {
if (arr[j]) {
if (flag!=-1) {
Sign(n, flag);
Sign(n, j);
return false;
}
flag = j;
}
j += (n - 1);
}
}
for (int i = 2*n-1; i <= n * (n - 1); i = i + n) {
int flag = -1;
int j = i;
while (j < n * n) {
if (arr[j]) {
if (flag!=-1) {
Sign(n, flag);
Sign(n, j);
return false;
}
flag = j;
}
j += (n - 1);
}
}
return true;
}
//在控制台展现棋盘状态
void ShowChessboard_cmd(int n, bool* arr) {
for (int i = 0; i < n*n; ++i) {
if (i % n == 0) std::cout << std::endl;
std::cout << arr[i] << " ";
}
std::cout << std::endl;
}
void solve_queen(int StartIndex, int n, bool* arr,std::vector<std::vector<bool>>& Ret) {
if (!Check(n, arr)) {
std::cout << "错误 "<<std::endl;
return;
}
//剪枝
if (StartIndex == n && Check(n, arr)) {
std::vector<bool> v;
for (int i = 0; i < n * n; ++i) {
v.push_back(arr[i]);
}
Ret.push_back(v);
return;
}
for (int i = StartIndex * n; i < StartIndex * n + n; ++i) {
arr[i] = true;
//展示状态
cleardevice();
BeginBatchDraw();
ShowChessboard(n, arr);
ShowChessboard_cmd(n, arr);
Check(n, arr);
FlushBatchDraw();
//Sleep(1);
solve_queen(StartIndex + 1, n, arr, Ret);
arr[i] = false;
}
}
void Show_Ret(std::vector<std::vector<bool>> Ret, int n) {
int num = Ret.size();
bool* arr = (bool*)malloc(sizeof(bool) * (n * n));
int i = 0;
for (std::vector<std::vector<bool>>::iterator it=Ret.begin();it<Ret.end();++it,++i) {
std::cout << "一共有 " << num << " 种答案" << std::endl;
std::cout << "这是第 " << i + 1 << " 个答案" << std::endl;
std::cout << "任意键切换下一个" << std::endl;
int j = 0;
for (std::vector<bool>::iterator it_2 = (*it).begin(); it_2 < (*it).end(); ++it_2,++j) {
arr[j] = *it_2;
}
cleardevice();
BeginBatchDraw();
ShowChessboard(n,arr);
ShowChessboard_cmd(n, arr);
FlushBatchDraw();
system("pause");
}
system("pause");
}
int main() {
initgraph(1000, 1000);
bool* arr = nullptr;
const int n = 8;
initialChessboard(n, arr);
std::vector<std::vector<bool>> Ret;
solve_queen(0, n, arr, Ret);
free(arr);
Show_Ret(Ret, n);
return 0;
}
运行图片:
红色代表出现错误的棋子
运行中:
运行结果: