题目描述
大楼的每一层楼都可以停电梯,而且第 i 层楼上有一个数字 Ki。
电梯只有四个按钮:开,关,上,下,上下的层数等于当前楼层上的那个数字。
当然,如果不能满足要求,相应的按钮就会失灵。
例如:3 3 1 2 5 代表了 Ki(K1 = 3, K2 = 3, ……),从一楼开始。
在一楼,按 “上” 可以到 4 楼,按 “下” 是不起作用的,因为没有 −2 楼。
那么,从 A 楼到 B 楼至少要按几次按钮呢?
输入格式
第一行为三个用空格隔开的正整数,表示 N, A, B;
第二行为 N 个用空格隔开的正整数,表示 Ki。
输出格式
一行,即最少按键次数,若无法到达,则输出 −1
。
输入样例
5 1 5
3 3 1 2 5
输出样例
3
数据范围
1 ≤ N ≤ 200
1 ≤ A, B ≤ N
题解
BFS:
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
const int N = 210;
int n, a, b;
int w[N], dist[N];
int bfs()
{
memset(dist, -1, sizeof dist);
queue<int> q;
q.push(a);
dist[a] = 0;
while(q.size())
{
int t = q.front();
q.pop();
if(t == b) return dist[b];
if(t + w[t] <= N && dist[t + w[t]] == -1)
{
q.push(t + w[t]);
dist[t + w[t]] = dist[t] + 1;
}
if(t - w[t] >= 1 && dist[t - w[t]] == -1)
{
q.push(t - w[t]);
dist[t - w[t]] = dist[t] + 1;
}
}
return -1;
}
int main()
{
cin >> n >> a >> b;
for (int i = 1; i <= n; i ++) cin >> w[i];
cout << bfs() << endl;
return 0;
}