luogu | 奇怪的电梯

题目背景

感谢 @yummy 提供的一些数据。

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 $i$ 层楼（$1\le i\le N$）上有一个数字 $K_i$（$0\le K_i\le N$）。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如： $3,3,1,2,5$ 代表了 $K_i$（$K_1=3$，$K_2=3$，……），从 $1$ 楼开始。在 $1$ 楼，按“上”可以到 $4$ 楼，按“下”是不起作用的，因为没有 $-2$ 楼。那么，从 $A$ 楼到 $B$ 楼至少要按几次按钮呢？

输入格式

共二行。

第一行为三个用空格隔开的正整数，表示 $N,A,B$（$1\le N\le 200$，$1\le A,B\le N$）。

第二行为 $N$ 个用空格隔开的非负整数，表示 $K_i$。

输出格式

一行，即最少按键次数，若无法到达，则输出 -1。

样例 #1

样例输入 #1

5 1 5
3 3 1 2 5

样例输出 #1

3

提示

对于 $100\%$ 的数据，$1\le N\le 200$，$1\le A,B\le N$，$0\le K_i\le N$。

本题共 $16$ 个测试点，前 $15$ 个每个测试点 $6$ 分，最后一个测试点 $10$ 分。

思路

我们可以使用bfs来寻找最短路径，因为在bfs中只要先到达，就必然是最短的（当然这也是仅仅是在计算“最少多少步才能到达的问题”上用的比较频繁）。

首先，我们要考虑一个事情，假如电梯在某个点的时候可以到哪里，代码如下

int fy1 = top.y + a[top.y];
int fy2 = top.y - a[top.y];

然后，就是限制条件了，我们要限制电梯不能超过楼层的范围，如果符合条件，就将这个楼层加入到队列中，代码如下

#define yBegin 1
#define yEnd N
if(fy1 <= yEnd && b[fy1] == 0) {
    Q.push((node){fy1, top.step+1});
    b[fy1] = 1;
}
if(fy2 >= yBegin && b[fy2] == 0) {
    Q.push((node){fy2, top.step+1});
    b[fy2] = 1;
}

至于，为什么要用b数组，是因为我们要记录电梯是否已经到达过这个楼层，如果到达过，就不需要再次到达了。

最后，如果到达了B楼，就输出结果，代码如下

if(top.y == B) {
    res = top.step;
    break;
}

Code

#include <bits/stdc++.h>
using namespace std;
#define yBegin 1
#define yEnd N

struct node {
	int y, step;
};

int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int N, A, B, res = -1;
	array<int, 1000> a, b;
	queue<node> Q;
	cin >> N >> A >> B;
	for(int i = 1; i <= N; ++i) {
		cin >> a[i];
		b[i] = 0;
	}
	Q.push((node){A, 0});   // 从A楼开始
	while(!Q.empty()) {
		node top = Q.front();
		Q.pop();
		if(top.y == B) {
			res = top.step;
			break;
		}
		if(a[top.y] == 0) continue;
		int fy1 = top.y + a[top.y], fy2 = top.y - a[top.y];
		if(fy1 <= yEnd && b[fy1] == 0) {
			Q.push((node){fy1, top.step+1});
			b[fy1] = 1;
		}
		if(fy2 >= yBegin && b[fy2] == 0) {
			Q.push((node){fy2, top.step+1});
			b[fy2] = 1;
		}
	}
	cout << res;
}