传送锚点: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;
}