奇怪的电梯

奇怪的电梯

题目描述

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

输入

第1行为三个正整数，表示N，A，B(1≤N≤200，1≤A，B≤N)；
第2行为N个正整数，表示Ki。

输出

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

样例输入

5 1 5
3 3 1 2 5

样例输出

3
分析：bfs，注意楼层id的限制；
代码：

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<map>
#include <queue>
#define ll long long
#define rep(i,m,n) for(i=m;i<=n;i++)
#define inf 0x3f3f3f3f
const int maxn=1e3+10;
using namespace std;
int n,m,a,b;
const int dis[][2]={0,1,1,0,-1,0,0,-1};
int q[maxn],vis[maxn];
int main()
{
    int i,j,k,t;
    queue<int>p;
    scanf("%d%d%d",&n,&a,&b);
    vis[a]=1;
    for(i=1;i<=n;i++)scanf("%d",&q[i]);
    p.push(a);
    while(!p.empty())
    {
        int now=p.front(),l=now-q[now],r=now+q[now];
        p.pop();
        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;
        if(l==b||r==b)break;
    }
    printf("%d\n",vis[b]-1);
    //system("pause");
    return 0;
}