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