大楼的每一层楼都可以停电梯,而且第 i 层楼 (1≤i≤N)上有一个数字 Ki(0≤Ki≤N)。
电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3 3 1 2 5 代表了 Ki(K1=3,K2=3,...),从一楼开始。在一楼,按“上”可以到 4 楼,按“下”是不起作用的,因为没有-2楼。那么,从 A 楼到 B 楼至少要按几次按钮呢?
输入描述:
输入共有二行。
第一行为三个用空格隔开的正整数,表示 N,A,B (1≤N≤200,1≤A,B≤N)。
第二行为 N 个用空格隔开的正整数,表示 Ki
输出描述:
输出仅一行,即最少按键次数,若无法到达,则输出 -1。
用例输入:
5 1 5
3 3 1 2 5
用例输出:
3
#include <bits/stdc++.h>
using namespace std;
queue<int>q;
int k[205] = {};
int mid[205] = {};
int main() {
int n, a, b; //n代表有多少层楼,a
cin >> n >> a >> b;
for (int i = 1; i <= n; i++) cin >> k[i];
for (int i = 1; i <= n; i++) mid[i] = 123456;
mid[a] = 0; //开始的位置是0步
q.push(a);//开始的位置进入队列
while (!q.empty()) {
int x = q.front(); //队首,就是现在的楼层
q.pop();//已经取出队首了,那么可以开始判它可以走到的范围了
// x->x+ k[x] x->x-k[x] 但是前提不能超出范围
int up = x + k[x];
int down = x - k[x];
if (up <= n and mid[up] == 123456) {
mid[up] = mid[x] + 1;
//cout<<"现在的楼层"<<up<<"需要走"<<mid[up]<<endl;
q.push(up);
if (up == b) break;
}
if (down > 0 and mid[down] == 123456) {
mid[down] = mid[x] + 1;
//cout<<"现在的楼层"<<down<<"需要走"<<mid[down]<<endl;
q.push(down);
if (down == b) break;
}
}
if (mid[b] != 123456)
cout << mid[b];
else
cout << -1;
return 0;
}