【C++】【洛谷】P1135 奇怪的电梯

奇怪的电梯

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 $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$。

代码思路

这么水的题竟然能评黄我是没有想到的。实际上就是经典题抓住那头牛的延伸拓展版，没啥好说的，上代码！

示例代码 Code

#include <bits/stdc++.h>
using namespace std;
struct Node {
    int pos, sum;
};
queue <Node> q;
int n, a, b, d[205];
bool v[205];
int main() {
    cin >> n >> a >> b;
    for (int i = 1; i <= n; i++) {
        cin >> d[i];
    }
    Node h, t;
    h.pos = a;
    h.sum = 0;
    v[a] = true;
    q.push(h);
    while (!q.empty()) {
        h = q.front();
        if (h.pos == b) {
            cout << h.sum << endl;
            return 0;
        }
        for (int i = 1; i <= 2; i++) {
            if (i == 1)
                t.pos = h.pos + d[h.pos];
            else 
                t.pos = h.pos - d[h.pos];
            t.sum = h.sum + 1;
            if (t.pos < 1 || t.pos > n) continue;
            if (!v[t.pos]) {
                v[t.pos] = true;
                q.push(t);
            }
        }
        q.pop();
    }
    cout << -1 << endl;
    return 0;
}