洛谷 P1135 奇怪的电梯（BFS）

P1135 奇怪的电梯

题目背景

感谢 @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$ 分。

题解

#include <iostream>
#include <queue>
#include <cstring>
#define Max 210
using namespace std;

int main() {
    int N, A, B;
    cin >> N >> A >> B;
    int K[Max];  // K[1]到K[N]存储各楼层的数字
    for (int i = 1; i <= N; ++i) {
        cin >> K[i];
    }

    bool visited[Max];  // 标记楼层是否被访问过
    memset(visited, false, sizeof(visited));
    queue<pair<int, int>> q;  // 队列，存储(当前楼层, 按键次数)
    q.push({ A, 0 });
    visited[A] = true;

    int result = -1;
    while (!q.empty()) {
        auto now = q.front();
        q.pop();
        int floor = now.first;
        int steps = now.second;

        if (floor == B) {  // 到达目标楼层
            result = steps;
            break;
        }

        // 尝试向上
        int upFloor = floor + K[floor];
        if (upFloor <= N && !visited[upFloor]) {
            visited[upFloor] = true;
            q.push({ upFloor, steps + 1 });
        }

        // 尝试向下
        int downFloor = floor - K[floor];
        if (downFloor >= 1 && !visited[downFloor]) {
            visited[downFloor] = true;
            q.push({ downFloor, steps + 1 });
        }
    }

    cout << result << endl;
    return 0;
}

C++ 标准库中的 std::pair 可以存储两个任意类型的数据，是处理成对数据的便捷选择

queue<pair<int, int>> q;