【基础】奇怪的电梯

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 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 。

代码

#include<iostream>
#include<queue>
using namespace std;
typedef struct
{
    int step;
    int floor;
} dt;
queue<dt>lift;
bool flag[2000],f;
int main()
{
    int n,a,b,k[2000];
    dt s,e;
    cin>>n>>a>>b;
    for(int i=1;i<=n;i++)
    {
        cin>>k[i];
    }
    s.step =0;s.floor =a;
    lift.push(s);
    while(!lift.empty())
    {
        s=lift.front();
        lift.pop() ;
        if(s.floor ==b)
        {
            cout<<s.step<<endl;
            f=1;
            break;
        }
        if(s.floor +k[s.floor]<=n&&!flag[s.floor+k[s.floor]])//如果楼层在范围内且没被走过 
        {
            e.floor =s.floor+k[s.floor];
            e.step =s.step +1;
            flag[e.floor]=1;
            lift.push(e); 
        }
        if(s.floor -k[s.floor]>=1&&!flag[s.floor-k[s.floor]])
        {
            e.floor =s.floor-k[s.floor];
            e.step =s.step +1;
            flag[e.floor]=1;
            lift.push(e); 
        }
    }
    if(f==0)    cout<<"-1"<<endl; 
    return 0;
}