题目描述
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第 i 层楼(1≤i≤N)上有一个数字 Ki(0≤Ki≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如: 3,3,1,2,5 代表了 Ki(K1=3,K2=3,……),从 1 楼开始。在 1 楼,按“上”可以到 4 楼,按“下”是不起作用的,因为没有 −2楼。那么,从 A 楼到 B 楼至少要按几次按钮呢?
输入格式
共二行。
第一行为三个用空格隔开的正整数,表示 N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为 N 个用空格隔开的非负整数,表示 Ki。
输出格式
一行,即最少按键次数,若无法到达,则输出 -1
。
输入输出样例
输入 #1复制
5 1 5 3 3 1 2 5
输出 #1复制
3
说明/提示
对于 100% 的数据,1≤N≤200,1≤A,B≤N,0≤Ki≤N。
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int n, st, ed, to[205];
bool vis[205];
struct node{
int id, step; /* id: 当前楼层 step: 按钮次数 */
} x;
queue <node> q;
int main ()
{
cin >> n >> st >> ed;
for (int i=1; i<=n; i++)
{
cin >> to[i];
}
q.push((node){st,0});
while (q.size())
{
x = q.front();
q.pop();
if (x.id==ed) break;
if (x.id+to[x.id] <= n && !vis[x.id+to[x.id]])
{
q.push((node){x.id+to[x.id],x.step+1});
vis[x.id+to[x.id]] = 1;
}
if (x.id-to[x.id] >= 1 && !vis[x.id-to[x.id]])
{
q.push((node){x.id-to[x.id],x.step+1});
vis[x.id-to[x.id]] = 1;
}
}
if (x.id==ed) cout << x.step;
else cout << -1;
return 0;
}