【ACM- OJ】《奇怪的电梯》C++

题目描述

有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第i层楼(1≤i≤N)上有一个数字Ki(0≤Ki≤N)。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如：3,3,1,2,5代表了Ki(K1=3,K2=3,…)，从1楼开始。在1楼，按“上”可以到4楼，按“下”是不起作用的，因为没有−2楼。那么，从A楼到B楼至少要按几次按钮呢？

输入

共二行。
第一行为3个用空格隔开的正整数，表示N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为N个用空格隔开的非负整数，表示Ki。

输出

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

样例输入

5 1 5
3 3 1 2 5

样例输出

3

AC代码

//
//  main.cpp
//  奇怪的电梯
//
//  Created by HISS on 2020/11/5.
//  Copyright © 2020 HISS. All rights reserved.
//


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

int n, m, a, b;
int q[250], vis[250];

int main()
{
    int i;
    queue<int> p;
    cin >> n >> a >> b;
    vis[a]=1;
    for(i = 1; i <= n; i ++)
        cin >> q[i];
    p.push(a);
    while(!p.empty())
    {
        int now = p.front(), l = now - q[now], r = now + q[now];
        //电梯向下
        if(l >= 1 && !vis[l])
        {
            p.push(l);
            vis[l] = vis[now] + 1;
        }
        //电梯向上
        if(r <= n && !vis[r])
        {
            p.push(r);
            vis[r] = vis[now] + 1;
        }
        //到达目标b
        if(l == b || r == b)
            break;
        p.pop();
    }
    cout << vis[b] - 1 << endl;
}