//int队列存储层数的bfs
#include<iostream>
#include<queue>
using namespace std;
const int N = 310;
int n, A, B;
queue<int>q;
int step;
bool vis[N];
int a[N];
int bfs() {
q.push(A);
vis[A] = true;
while (++step) { //每次对所有对头枚举过后,也就是要走下一步的时候,把走过的步数加一
if (q.empty()) { //如果队列已经空了,就说明无法到达目标楼层
return -1;
}
int len = q.size(); //在此处先行存储下当前的队列长度,因为下面循环里会使队列长度变化
for (int i = 0; i < len; i++) { //循环遍历取出上一次压入的所有层数(不包含在枚举过程中压入的下一次枚举的层数)
int t = q.front();
q.pop();
//枚举出对头往上走和往下走的情况
if (t + a[t] <= n) { //如果往上走是合法的
if (!vis[t + a[t]]) { //如果这个状态没有经历过
if (t + a[t] == B) { //如果现在已经到达了目标层数
return step;
}
vis[t + a[t]] = true; //标记上此状态
q.push(t + a[t]); //把这一层数压入队列,以进行下一次的层数枚举
}
}
if (t - a[t] >= 1) { //如果往下走是合法的
if (!vis[t - a[t]]) {
if (t - a[t] == B) {
return step;
}
vis[t - a[t]] = true;
q.push(t - a[t]);
}
}
}
}
return -1;
}
int main() {
cin >> n >> A >> B;
if (A == B) { //如果从一开始,起始楼层就等于目标楼层,那么直接输出0步
printf("0");
return 0;
}
for (int i = 1; i <= n; i++) {
cin >> a[i];
}
int ans = bfs();
printf("%d", ans);
return 0;
}
洛谷 P1135 奇怪的电梯
于 2023-12-27 22:33:35 首次发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)