洛谷 P1135 奇怪的电梯

P1135 奇怪的电梯

代码实现

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;

// 定义队列 结构体
// 使用结构体作为队列元素
typedef struct
{
    int floor; // 当前楼层编号
    int step;  // 到达该层所需要的步数
} QElement;

queue<QElement> q; // 定义元素类型为QElement的队列q

const int N = 300;

int n, a, b;
int s[N]; // 记录每个楼层可以上下的楼层数
int t[N]={0}; // 记录各个楼层是否到达过

void bfs(int a)
{
    int i;
    QElement e1, e2; // 创建队列

    e1.floor = a;
    e1.step = 0;

    q.push(e1);             // 起始位置入队，当前楼层为a，经历过0次
    t[a] = 1;               // 已经到达过为0
    while (!q.empty())
    {
        e2 = q.front();    // 获取队头元素
        q.pop();           // 队头元素出队
        if (e2.floor == b) // 如果当前楼层为b，之间跳出循环
            break;

        // 向上走
        i = e2.floor + s[e2.floor]; // 按向上按钮后能够到达的楼层
        if (i <= n && t[i] == 0)    // 如果按向上按钮能到达的楼层有效并且未访问过该楼层
        {
            e1.floor = i;
            e1.step = e2.step + 1;
            q.push(e1);
            t[i] = 1; //设该楼层为已访问过
        }

        // 向下走
        i = e2.floor - s[e2.floor]; // 按向下按钮后能够到达的楼层
        if (i >= 1 && t[i] == 0)    // 如果按向下按钮能到达的楼层有效并且未访问过该楼层
        {
            e1.floor = i;
            e1.step = e2.step + 1;
            q.push(e1);
            t[i] = 1; //设该楼层为已访问过
        }
    }
    //如果当前楼层为b，输出按按钮次数，否则无解（输出-1）
    if (e2.floor == b)
        cout << e2.step << endl;
    else
        cout << -1 << endl;
}

int main()
{
    cin >> n >> a >> b;
    for (int i = 1; i <= n; i++)
        cin >> s[i];

    bfs(a);

    return 0;
}