奇怪的电梯
传送锚点:https://www.luogu.com.cn/problem/P1135
题目描述
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第 \(i\) 层楼(\(1 \le i \le N\))上有一个数字 \(K_i\)(\(0 \le K_i \le N\))。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如: \(3, 3, 1, 2, 5\) 代表了 \(K_i\)(\(K_1=3\),\(K_2=3\),……),从 \(1\) 楼开始。在 \(1\) 楼,按“上”可以到 \(4\) 楼,按“下”是不起作用的,因为没有 \(-2\) 楼。那么,从 \(A\) 楼到 \(B\) 楼至少要按几次按钮呢?
输入格式
共二行。
第一行为三个用空格隔开的正整数,表示 \(N, A, B\)(\(1 \le N \le 200\),\(1 \le A, B \le N\))。
第二行为 \(N\) 个用空格隔开的非负整数,表示 \(K_i\)。
输出格式
一行,即最少按键次数,若无法到达,则输出 -1。
样例 #1
样例输入 #1
5 1 5
3 3 1 2 5
样例输出 #1
3
提示
对于 \(100 \%\) 的数据,\(1 \le N \le 200\),\(1 \le A, B \le N\),\(0 \le K_i \le N\)。
本题共 \(16\) 个测试点,前 \(15\) 个每个测试点 \(6\) 分,最后一个测试点 \(10\) 分。
思路
此题我们要优化我们的剪枝方案,引入ans统计从起点到该层最小按键次数,若cnt + 1大于等于到达该层的最小按键次数,直接剪枝,进行优化
code
#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 210;
int n,a,b;
int evlt[maxn];//存储电梯初始数字
int res = 1e9;//最少按键次数
int ans[maxn];//记录从起点到此层的最小按键次数
void dfs(int x, int cnt) {//x代表第几层,cnt为按的次数
ans[x] = cnt;
//上
if (x - evlt[x] > 0 && cnt + 1 < ans[x - evlt[x]] ) {
dfs(x - evlt[x], cnt + 1);
}
//下
if (x + evlt[x] <= n && cnt + 1 < ans[x + evlt[x]]) {
dfs(x + evlt[x], cnt + 1);
}
}
int main()
{
cin >> n >> a >> b;
memset(ans,0x7f,sizeof(ans));//0x7f为127,作为初始化状态
for (int i = 1; i <= n; i++) {
cin >> evlt[i];
}
dfs(a, 0);
//ans整数数组,每个整数占用4个字节,初始化时,每个整数的每个字节都会被设置为0x7f,
if(ans[b] != 0x7f7f7f7f){//判断ans[b]是否被计算过
cout << ans[b];
}
else cout << -1;
return 0;
}
浙公网安备 33010602011771号