洛谷 P1135 奇怪的电梯-普及/提高-

智趣代码实验室

于 2025-08-06 20:15:00 发布

阅读量489

点赞数 6

CC 4.0 BY-SA版权

分类专栏：洛谷文章标签： c++ 算法开发语言洛谷

本文链接：https://blog.youkuaiyun.com/weixin_37253733/article/details/149940277

洛谷专栏收录该内容

96 篇文章

订阅专栏

题目背景

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

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 $i$ 层楼（ $\le i \le N$ ）上有一个数字 $K_i$ （ $\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$ （ $\le N \le 200$ ， $\le A, B \le N$ ）。

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

输出格式

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

输入输出样例 #1

输入 #1

5 1 5
3 3 1 2 5

输出 #1

说明/提示

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

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

solution

用 bfs 从起点楼层开始，每个点可以有两种状态，都尝试一下，加入队列，继续尝试，直到 B 楼出现搜索完毕

代码

#include <sstream>
#include "iostream"
#include "algorithm"
#include "vector"
#include "unordered_map"
#include "unordered_set"
#include "queue"

using namespace std;

const int N = 201;
int K[N], a[N];

int main() {
    int n, A, B;
    cin >> n >> A >> B;
    for (int i = 1; i <= n; i++) cin >> K[i];
    queue<int> q;
    q.push(A);
    a[A] = 1;
    while (!q.empty()) {
        int c = q.front();
        if (c == B) {
            cout << a[c] - 1;
            return 0;
        }
        q.pop();
        int x = c + K[c];
        if (x <= n && !a[x]) {
            q.push(x);
            a[x] = a[c] + 1;
        }
        int y = c - K[c];
        if (y >= 1 && !a[y]) {
            q.push(y);
            a[y] = a[c] + 1;
        }
    }

    cout << "-1";
    return 0;
}