洛谷-1135 奇怪的电梯

题目描述
呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第iii层楼(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个用空格隔开的非负整数，表示K。
输出格式：
一行，即最少按键次数,若无法到达，则输出−1。

输入输出样例
输入样例#1：
5 1 5
3 3 1 2 5

输出样例#1：
3

解释：最短路问题，每层一个节点i，然后连接i+a[i],i-a[i]点，注意判断是否存在，然后后就是跑A到B的最短路了

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 603
#define INF 100000009
using namespace std;
int a[N]={0},A=0,B=0;
int n=0;
int head[N]={0};
int next1[N]={0};
int V[N]={0};
int dist[N]={0};
int tot=0;
void add(int x,int y){
    tot++;
    next1[tot]=head[x];
    V[tot]=y;head[x]=tot;
}
bool check(int x){
    if(x<=n&&x>=1) return 1;
    return 0;
}
void ok(int A){
    fill(dist,dist+n+1,INF);
    dist[A]=0;
    for(int k=1;k<=n;k++){
        for(int i=1;i<=n;i++){
            for(int j=head[i];j;j=next1[j]){
                int to=V[j],w=1;
                if(dist[to]>dist[i]+w){
                    dist[to]=dist[i]+1;
                }
            }
        }
    }
}
int main(){
    scanf("%d%d%d",&n,&A,&B);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++){
        if(check(i+a[i])) add(i,i+a[i]);
        if(check(i-a[i])) add(i,i-a[i]);
    }
    ok(A);
    if(dist[B]>=INF) cout<<"-1"<<endl;
    else cout<<dist[B]<<endl;
    return 0;
}