HDU 1548【奇怪的电梯】

描述

　　呵呵，有一天ssxyh做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第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

输入输出样例

输入样例1

5 1 5
3 3 1 2 5

输出样例1

3

解题思路

　　这道题还是比较水的，每次进队列，再做上升和下降操作，并判断越界即可。

题解

#include<bits/stdc++.h>
using namespace std;
int n,a,b; 
int k[100001];//移动层数 
int flag[100001];//到达的最少步数 
queue<int> q;
void bfs()
{
    q.push(a);
    while(!q.empty())
    {
        int head=q.front();//取队首元素 
        q.pop();//弹出 
        if(head==b)//层数到达 
        {
            cout<<flag[head];///输出 
            return;
        }
        if(flag[head+k[head]]==-1&&head+k[head]<=n)//上升，判断是否到达过并且越界 
        {
            flag[head+k[head]]=flag[head]+1;
            q.push(head+k[head]);
        }
        if(flag[head-k[head]]==-1&&head-k[head]>=1)//下降 ，判断是否到达过并且越界
        {
            flag[head-k[head]]=flag[head]+1;
            q.push(head-k[head]);
        }
    }
    cout<<flag[b];//没到达输出-1 
}
int main()
{
    cin>>n>>a>>b;
    for(int i=1;i<=n;i++)
    {
        cin>>k[i];//输入移动层数 
        flag[i]=-1;//全部赋初值为-1 
    }
    flag[a]=0;//从a开始，所以是0 
    bfs();
    return 0;
}