HDU1548 A strange lift BFS 简单题

一个电梯有n层，每一层有一个数k[i],和2个按钮，UP和DOWN，表示从这一层可以到达i+k[i] 或i-k[i] .

给出a,b,问从a到b 最少需要按多少下按钮。

直接bfs.

#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
const int maxn=210;
int n,a,b;
int k[maxn];
bool vis[maxn];
int time[maxn];
int bfs()
{
    memset(vis,false,sizeof(vis));
    queue<int>que;
    while(!que.empty())
        que.pop();
    que.push(a);
    vis[a]=true;
    time[a]=0;
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        if(u==b)
            return time[b];
        if(u-k[u]>0&&!vis[u-k[u]])
        {
            que.push(u-k[u]);
            vis[u-k[u]]=true;
            time[u-k[u]]=time[u]+1;
        }
        if(u+k[u]<=n&&!vis[u+k[u]])
        {
            que.push(u+k[u]);
            vis[u+k[u]]=true;
            time[u+k[u]]=time[u]+1;
        }
    }
    return -1;
}
int main()
{
    while(scanf("%d%d%d",&n,&a,&b))
    {
        if(!n)
            break;
        for(int i=1;i<=n;i++)
            scanf("%d",&k[i]);
        int ans=bfs();
        printf("%d
",ans);
    }
    return 0;
}