23-2-29回溯法

Agnes_A20

已于 2024-04-10 18:43:45 修改

阅读量515

点赞数 9

文章标签：算法数据结构

于 2024-02-29 18:16:07 首次发布

本文链接：https://blog.csdn.net/2301_79611133/article/details/136376687

版权

本文介绍了使用C++解决八皇后问题的代码，并通过改进策略优化了程序效率，主要关注行、列和对角线的逻辑，以及使用位向量法进行标记和复原。

摘要由CSDN通过智能技术生成

代码借鉴刘汝佳《算法竞赛入门经典》，萌新自用

1.八皇后

void search(int cur){//cur为当前皇后数
	if(cur==n)tot++;
	else for(int i=0;i<n;i++){
		int ok=1;
		c[cur]=i;//按行递增排皇后，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;//副对角线计算cur-c[cur]==j-c[j]
				break;
			}
		}
		if(ok)search(cur+1);
	}
}

提高程序效率

思路：i为行，cur为列和皇后序号，cur+i代表右上到左下的对角线，cur-i+n为另一条对角线

#include <stdio.h>
#define MAXN 10
int n = 8; 
int c[MAXN] = {0}; // 存放每行皇后的列号
int vis[3][MAXN * 2] = {0}; // 标记数组
int tot = 0; // 计数器

void search(int cur){
    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]){
                c[cur] = i;
                vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 1;
                search(cur+1);
                vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 0;
                //第一次遍历完，解答树已有n层，n层点恢复后回到n-1层继续遍历，直到满足条件方向向下
            	//而位向量法复原和搜索都在循环外的原因是其解答树为各层组合式，该层组合完就往下组合，到底后回到最高层
			}
        }
    }
}

int main(){
    search(0); // 调用搜索函数
    printf("Total solutions: %d\n", tot); // 输出结果
    return 0;
}