HDU1548A strange lift(BFS)

HDUOJ1548
题意：有一个电梯，每一层规定一个数字k，只能按键从这一层走到这一层+k或-k层，给出初始楼层和目的楼层输出按键最少次数
不能到达的话输出-1
思路：BFS…没的说，看代码
代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
struct node
{
    int floor;//楼层
    int step;//步数
};
int k[250];//每层的那个数字
int vis[250];//是否遍历过
int go[2]={-1,1};//方向上或下
int n,a,b;
int cnt;
queue<node> q;
node pre;//临时的存些东西用
node u;//这个也是临时的存些东西用
int bfs()
{
    q.push(node{a,0});//将起点和步数零这个node放到队首
    while(!q.empty())
    {
        u=q.front();//记下队首元素
        //cout<<u.floor<<' '<<u.step<<endl;//就是测试一下程序的一行代码，emmmmmmdebug很有用
        q.pop();//队首元素出队
        if(u.floor==b)//若到达目的地啦
        {
            return u.step;//返回步数
        }
        for(int i=0;i<2;i++)//遍历这一层的可上或下到的两个楼层
        {
            pre.floor=u.floor+k[u.floor]*go[i];
            pre.step=u.step+1;//步数加一
            if(pre.floor>=1&&pre.floor<=n&&!vis[pre.floor])//层数不超过限制并且未遍历过
                {
                    vis[pre.floor]=1;//标记
                    q.push(pre);//将这一层放到队首
                }
        }
    }
    return -1;
}

int main()
{
    while(cin>>n&&n!=0)//输入0代表输入的结束
    {
       cin>>a>>b;
        memset(vis,0,sizeof(vis));
        memset(k,0,sizeof(k));
        while(!q.empty())
            q.pop();//各种初始化

        for(int i=1;i<=n;i++)
            cin>>k[i];//输入每层的那个数
            cnt=bfs();
        cout<<cnt<<endl;
    }
    return 0;
}

另外emmmm不要落下走不过去的情况啊…最短路什么的很有可能用bfs