HDU 1548 A strange lift 奇怪的电梯(BFS，水)

 

题意：

　　有一座电梯，其中楼层从1～n，每层都有一个数字k，当处于某一层时，只能往上走k层，或者下走k层。楼主在a层，问是否能到达第b层？

 

思路：

　　在起点时只能往上走和往下走两个选择，之后的每层都是这样，那么就类似于二叉树。每个节点就是对应的层，因为有可能碰到循环的层，比如1跳到3,3跳回1,这样使得无限循环，所以加个vis数组标记是否遍历过即可。

 

#include <iostream>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <deque>
using namespace std;
const int N=2008;
int n, a, b;
int q[N];
deque<int> que;
bool vis[N];

int cal()
{
    if(a==b)    return 0;
    que.clear();


    int cnt=0;
    que.push_back(a);
    while(!que.empty())
    {
        int siz=que.size();
        for(int i=0; i<siz; i++)
        {
            int cur=que.front();
            que.pop_front();
            if(cur+q[cur]==b||cur-q[cur]==b)
                return cnt+1;

            if(cur+q[cur]<=n && !vis[cur+q[cur]] )    //没有遍历过的,小于上限的才考虑
            {
                que.push_back(cur+q[cur]);
                vis[cur+q[cur]]=1;
            }
            if(cur-q[cur]>0 && !vis[cur-q[cur]] )    //没有遍历过的，大于0的才考虑
            {
                que.push_back(cur-q[cur]);
                vis[cur-q[cur]]=1;
            }
        }
        cnt++;
    }
    return -1;
}


int main()
{
    //freopen("input.txt", "r", stdin);

    while(cin>>n,n)
    {
        memset(q,0,sizeof(q));
        memset(vis,0,sizeof(vis));

        cin>>a>>b;
        for(int i=1; i<=n; i++)
            cin>>q[i];

        printf("%d\n",cal());
    }

    return 0;
}