题目背景
感谢 @yummy 提供的一些数据。
题目描述
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第 i 层楼(1≤i≤N)上有一个数字 K[i](0≤K[i]≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如: 3,3,1,2,5 代表了 K[i](K[1]=3,K[2]=3,......),从 1 楼开始。在 1 楼,按"上"可以到 4 楼,按"下"是不起作用的,因为没有 −2 楼。那么,从 A 楼到 B 楼至少要按几次按钮呢?
输入格式
共二行。
第一行为三个用空格隔开的正整数,表示 N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为 N 个用空格隔开的非负整数,表示 K[i]。
输出格式
一行,即最少按键次数,若无法到达,则输出 -1
。
输入输出样例
输入 #1
5 1 5
3 3 1 2 5
输出 #1
3
说明/提示
对于 100% 的数据,1≤N≤200,1≤A,B≤N,0≤K[i]≤N。
本题共 16 个测试点,前 15 个每个测试点 6 分,最后一个测试点 10 分。
思路
从 A 开始,搜索每一条路,可以更新答案再继续搜索,不然会搜索多余导致 TLE(超时)。
AC Code
cpp
#include<bits/stdc++.h>
using namespace std;
int n,a,b,k[201],dis[201];
void dfs(int node,int step){
dis[node]=step;
int v=node-k[node];
if(1<=v&&step+1<dis[v])
dfs(v,step+1);
v=node+k[node];
if(v<=n&&step+1<dis[v])
dfs(v,step+1);
return;
}
int main(){
memset(dis,0x3f,sizeof(dis));
cin>>n>>a>>b;
for(int i=1;i<=n;i++)
cin>>k[i];
dfs(a,0);
cout<<(dis[b]==0x3f3f3f3f?-1:dis[b]);
return 0;
}