奇怪的电梯

奇怪的电梯

题目描述:

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

输入格式:

每组数据，共有二行，第一行为三个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N)，第二行为N个用空格隔开的正整数，表示Ki。

输出格式:

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

样例输入:

5 1 5
3 3 1 2 5

样例输出:

3

时间限制:1000ms
空间限制:128MByte

#include<bits/stdc++.h>
using namespace std;

int main()
{
    queue<int> q;
    bool p[205];
    int n, a, b, ans = -1, ar[205];
    cin>>n>>a>>b;
    for(int i = 1; i <= n; i++) scanf("%d", &ar[i]);
    q.push(a), q.push(0), p[a] = 1;
    while(!q.empty()){
        int n1 = q.front();
        q.pop();
        int s = q.front();
        q.pop();
        if(n1 == b){
            ans = s;
            break;
        }
        if(n1 + ar[n1] > 0 && n1 + ar[n1] <= n && !p[n1 + ar[n1]])
            q.push(n1 + ar[n1]), q.push(s +1), p[n1 + ar[n1]] = 1;
        if(n1 - ar[n1] > 0 && n1 - ar[n1] <= n && !p[n1 - ar[n1]])
            q.push(n1 - ar[n1]), q.push(s +1), p[n1 - ar[n1]] = 1;
    }
    cout<<ans<<endl;
    return 0;
}