递归+回溯经典问题----八皇后问题(参考紫书)

爱编程的大李子

于 2021-06-18 20:15:22 发布

阅读量259

点赞数

分类专栏：算法学习知识点总结文章标签：算法数据结构

本文链接：https://blog.csdn.net/LXYDSF/article/details/118032880

版权

算法学习知识点总结专栏收录该内容

31 篇文章 3 订阅

订阅专栏

问题描述

在一个8*8的棋盘上放置八个皇后 , 使得他们互相不攻击(皇后攻击范围为同行同列同对角线)

方法一 :

从64个格子中选一个子集 , 使得 " 子集中恰好有八个元素 , 且任意选出的两个格子都不是同一行,同一列同,一对角线" ,
这是子集枚举问题 , 然而 , 64个格子的自己有2^64个 , 所需处理数据过大 !

方法二:

从64个格子中选八个格子 , 称为组合生成问题 , 根据组合数学有4.426*10^9中方案 , 虽然比第一种好 , 但是数据量依旧大.

最终方法

我们发现每一行每一列恰好会放一个皇后 , 所以可以从第一行开始放 , 然后考虑第二行 , 依次进行下去 !这样就变成了全排列生成问题 , 这样的排列有 8! = 40320个 , 枚举量不会超过该数字
然而如果每次都枚举这么多次的话也会超时的 , 所以我们可以采用回溯+递归的方法.

实现代码1(递归)

#include<iostream>
using namespace std;

int C[50],tot = 0,n = 8,nc = 0;//nc:递归的次数. 

void search(int cur)//cur:第几行
{
	int i,j;
	nc++;
	if(cur==n){//如果皇后放置完,则结束. 
		tot++;
	}else{
		for(int i = 0; i < n;i++){//依次将皇后尝试放在该行的每一列 
			int ok=1;//用于标记该位置是否可行. 
			C[cur] = i;
			for(int j = 0; j < cur; j++){
				if(C[cur]==C[j]||C[cur]-cur==C[j]-j||C[cur]+cur==j+C[j]) {//判断是否纵向  和 斜向(主对角线和次对角线) 攻击 
					ok = 0;
					break;//如果 
				}
			} 
			if(ok){//  递归回来后 就会尝试在下一个位置放置.. 
				search(cur+1);//在下一行放置皇后. 
			} 
		}
	}
	
	
} 

int main()
{
	scanf("%d",&n);
	search(0);
	printf("%d\n",tot);
	printf("%d\n",nc);
	return 0;
 }

实现代码2(递归+回溯)

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;

int C[50],vis[3][50],tot = 0,n = 0,nc = 0;//vis用于 vis[0]标记列是否已放置皇后  vis[1] 用于标记副对角线是否已存在节皇后点.  vis[2]标记主对角线上是否存在皇后节点. 
//C:用于存放皇后的位置,如果不需要保存则可以忽略. 
void search(int cur){
	nc++;//递归次数统计 
	if(cur==n){
		tot++;
	}else{
		for(int i = 0;i < n; i++){
			if(!vis[0][i]&&!vis[1][cur+i]&&!vis[2][cur-i+n]){//由于 横纵坐标相减可能出现赋值所以加上n 虽然有负值但不会出现重复的现象. 
				C[cur] = i;// 保存皇后节点. 
				vis[0][i] = vis[1][cur+i]=vis[2][cur-i+n] = 1;//更新vis 
				search(cur+1); // 继续进行摆放下一个皇后节点   
				//为啥这里需要回溯呢?
				//因为当递归回到这里说明沿着之前的继续走已经没有路了,我们得去掉这个标记重新做出选择,即用C[cur]尝试其他的值...  
				vis[0][i] = vis[1][cur+i]=vis[2][cur-i+n] = 0;//进行回溯,看当前行是否可以 在其他位置摆放皇后. 
			}
		}
	} 
	
}

int main()
{
	scanf("%d",&n);
	memset(vis,0,sizeof(vis));
	search(0);//从第一行开始进行递归
	printf("%d\n",tot);
	printf("%d\n",nc); 
	return 0;
 }