P1135 奇怪的电梯

题目描述

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

输入格式

共二行。

第一行为33个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N)N,A,B(1≤N≤200,1≤A,B≤N)。

第二行为NN个用空格隔开的非负整数，表示K_iKi​。

输出格式

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

输入输出样例

输入 #1复制

5 1 5
3 3 1 2 5

输出 #1复制

3
懒起来了，就不写思路了

#include<iostream>
#include<queue>
#define N 666
using namespace std;
struct floor{
    int x/*记录层数*/,y/*记录按钮次数*/;
};
queue<floor>q;
int ski[N],n,a,b;
bool flo[N];
int main()
{
    cin >> n >> a >> b;
    for(int i=1;i<=n;i++)
        cin >> ski[i];
    floor tmp={a,0};
    q.push(tmp);
    flo[a]=true;
    while(!q.empty())
    {
        tmp=q.front();
        q.pop();
        if(tmp.x==b)
        {
            cout << tmp.y << endl;
            return 0;
        }
        for(int sign=-1;sign<=1;sign+=2)
        {
            int num=tmp.x+sign*ski[tmp.x];
            if(num>=1&&num<=n&&flo[num]==0)
            {
                floor cnt={num,tmp.y+1};
                q.push(cnt);
                flo[num]=true;
            }
        }
    }
    cout << -1 << endl;
    return 0;
}