奇怪的电梯（信息学奥赛一本通-T1360）

【题目描述】

大楼的每一层楼都可以停电梯，而且第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

【源程序】

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<cstdlib>
#include<queue>
#include<vector>
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define N 301
#define MOD 123
#define E 1e-6
using namespace std;
int dp[N],k[N];
int main()
{
    int n,a,b;
    bool flag=true;

    cin>>n>>a>>b;
    for(int i=1;i<=n;i++)
        cin>>k[i];
        
    dp[a]=1;
    while(flag)
    {
        flag=false;
        for(int i=1;i<=n;i++)
        {
            if(dp[i]>0)
            {
                if(i-k[i]>0)
                {
                    if(dp[i-k[i]]==0||dp[i-k[i]]>dp[i]+1)
                    {
                        dp[i-k[i]]=dp[i]+1;
                        flag=true;
                    }
                }
                if(i+k[i]<=n)
                {
                    if(dp[i+k[i]]==0||dp[i+k[i]]>dp[i]+1)
                    {						
                        dp[i+k[i]]=dp[i]+1;
                        flag=true;
                    }
                }
            }
        }
    }
    cout<<dp[b]-1<<endl;

    return 0;
}