奇怪的电梯——DFS算法

题目

题解

每到一层楼都面临了两种选择:上还是下?因此我们可以定义一个布尔数组用来记录选择。

终止条件其实也明显,要么到了B层,要么没有找到楼层。

如果找到了,选择一个步骤少的方式。又怎么表示没有找到楼层?

定义一个答案res,把它定义很大,如果找到了答案,更新最小值即可。

int res=1e9;

res = min(res,cnt);

这其实都很简单,关键在于怎么剪枝?

第一,找答案的过程中如果发现cnt>res,就可以剪掉了

第二,每一层尽可能只来一次,如果选择了相同的选择。就消耗了一定的cnt,不可取。如果选择了不同的方式,没找到B层也还行,如果找到了,那我们第一次到这个楼层的时候就该选这个不同的方式,整个方案就已经没有必要再进行下去了。所以综上,再选择的时候我们应该一个楼层最多走一次。

答案

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

const int N = 210;

int n;
int A,B;
int K[N];
int res = 1e9;
bool st[N] = {false};

void dfs(int x,int cnt)
{
	if(cnt > res) return ;
	if(x == B)
	{
		res = min(res,cnt);
		return ;
	}
	
	if(x+K[x] <= n && !st[x+K[x]]){
		st[x] = true;
		dfs(x+K[x],cnt+1);
		st[x]=false;
	}
	if(x-K[x] > 0 && !st[x+K[x]])
	{
		st[x] = true;
		dfs(x-K[x],cnt+1);
		st[x] = false;
	}
	
}

int main()
{
	cin >> n >> A >> B;
	for(int i = 1;i<=n;i++)
	{
		cin >> K[i];
	}
	
	st[A]=true;
	dfs(A,0);
	if(res == 1e9){
		cout << "-1";
		return 0;
	}
	cout << res;
	return 0;
}
相关推荐
cur1es1 小时前
数据结构Java--8
java·数据结构·算法·散列表
tainshuai1 小时前
朴素贝叶斯:用 “概率思维” 解决分类问题的经典算法
算法·分类·数据挖掘
Y200309163 小时前
支持向量机核心知识总结
算法·机器学习·支持向量机
小巫程序Demo日记4 小时前
插入排序讲解
数据结构·算法·排序算法
CoovallyAIHub4 小时前
应对不平衡数据集:MixUp、CutMix与Focal Loss实战指南
深度学习·算法·计算机视觉
NAGNIP5 小时前
一文理解提示微调(Prefix Tuning/Prompt Tuning/P Tuning)
算法
薛定谔的算法5 小时前
深入探索 ES6 中的 Map 数据结构
前端·javascript·算法
CoovallyAIHub5 小时前
AI如何一眼看穿鱼群健康?看改进HRNet模型实现水下健康监测
深度学习·算法·计算机视觉
稳兽龙5 小时前
codeforces(1045)(div2)D. Sliding Tree
c++·算法··路径树