洛谷刷题C++语言 | P1135 奇怪的电梯

学习C++从娃娃抓起！记录下洛谷C++学习和备考过程中的题目，记录每一个瞬间。

附上汇总贴：洛谷刷题C++语言 | 汇总_热爱编程的通信人的博客-CSDN博客

【题目描述】

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 i 层楼（1≤i≤N）上有一个数字 Ki（0≤Ki≤N）。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如： 3,3,1,2,5 代表了 Ki（K1=3，K2=3，……），从 1 楼开始。在 1 楼，按“上”可以到 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;
int n, a, b, tmp;
struct node {
    int up, down, visit;
}f[205];
int main()
{
    cin >> n >> a >> b;
    for (int i=1; i<=n; i++) {
        cin >> tmp;
        f[i].up = i+tmp;
        f[i].down = i-tmp;
        f[i].visit = -1;
    }
    queue<node> q;
    f[a].visit = 0;
    node tp = {f[a].up, f[a].down, f[a].visit};
    q.push(tp);
    while (!q.empty()) {
        tp = q.front();
        q.pop();
        if (tp.up>=1 && tp.up<=n && f[tp.up].visit==-1) {
            f[tp.up].visit=tp.visit+1;
            node t = {f[tp.up].up, f[tp.up].down, f[tp.up].visit};
            q.push(t);
        }
        if (tp.down>=1 && tp.down<=n && f[tp.down].visit==-1) {
            f[tp.down].visit=tp.visit+1;
            node t = {f[tp.down].up, f[tp.down].down, f[tp.down].visit};
            q.push(t);
        }
        if (tp.up == b || tp.down == b) break;
    }
    cout << f[b].visit << endl;
    return 0;
}

【运行结果】

5 1 5
3 3 1 2 5
3