奇怪的电梯——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;
}
相关推荐
椎名ひる20 分钟前
2140、解决智力问题
数据结构·算法·leetcode
25405465202 小时前
710SJBH基于Apriori算法的学籍课程成绩关联规则挖掘研究
大数据·算法·spark
hao_wujing2 小时前
CNN工作原理和架构
算法
钢铁男儿3 小时前
C# 数组(数组是对象和一维数组和矩形数组)
算法·c#
W说编程7 小时前
算法导论第二十四章 深度学习前沿:从序列建模到创造式AI
c语言·人工智能·python·深度学习·算法·性能优化
寒燕舞8 小时前
2012 - 正方形矩阵
数据结构·算法
hqxstudying10 小时前
深入解析 Java List 实现类的底层原理
java·数据结构·算法·链表
C羊驼12 小时前
C语言:排序算法
c语言·算法·排序算法
float_com12 小时前
【单调栈】-----【Largest Rectangle in a Histogram】
算法··单调栈
↣life♚12 小时前
SAM2论文解读-既实现了视频的分割一切,又比图像的分割一切SAM更快更好
人工智能·深度学习·算法·计算机视觉·视频分割·通用分割