HDU 1548 A strange lift

原题链接：

http://acm.hdu.edu.cn/showproblem.php?pid=1548

题目大意：

输入N,A,B。分别表示有N层楼，最初楼层为A。终点楼层为B。然后是N个数表示在第i层楼按上或者下可以移动多少层。楼层不能超过N也不能低于1.
例如：
5 1 5
3 3 1 2 5
表示有5层楼，起点为1.终点为5。在3楼时。按下按钮可以上1层楼，或者下1层楼。

思路：

采用一般的bfs的方法既可以解得
一个数组存放访问标记是否到过这层楼。
一个数组存放到达这层楼的前一层楼是哪一层
越界判定和输出判定

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;

int N,A,B;
int Floor[205];
int Check[205];
int Pre[205];
int number;
int Up_Down[2]={ 1, -1 };
bool Can;

bool bIsLeagal(int Next )
{
    if( Next < 1 || Next > N )
        return false;
    return true;
}

void GetCount(int Index)
{
    while( Pre[ Index ] != -1 )
    {
        Index = Pre[ Index ];
        number++;
    }

}
void bfs()
{
    queue<int> check_floor;
    memset( Check,0,sizeof(Check));
    memset( Pre,0,sizeof( Pre ));
    Pre[A] = -1;
    Check[A] = true;
    check_floor.push ( A );

    int Now,Next;
    while( !check_floor.empty ())
    {
        Now = check_floor.front ();
        check_floor.pop ();
        if( Now == B )
        {
            Can = true;
            GetCount( Now );
            return;
        }
        int i;
        for( i = 0; i < 2; i++ )
        {
            Next = Now + Floor[Now] * Up_Down[i];
            if( bIsLeagal( Next ) && !Check[Next] )
            {
                Pre[Next] = Now;
                Check[Next] = true;
                check_floor.push ( Next );
            }

        }
    }
}
int main()
{
    while( cin >> N  )
    {
        if( N == 0 ) break;
        cin >> A >> B;
        Can = false;
        number = 0;
        memset( Floor,0,sizeof( Floor));
        int i;
        for( i = 1; i <= N; i++ )
            cin>> Floor[i];
        bfs();
        if( Can ) 
            cout<< number <<endl;
        else 
            cout<<-1<<endl;
    }
    return 0;
}