BFS: 奇怪的电梯

BFS

题目描述

有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第i层楼(1≤i≤N)上有一个数字Ki(0≤Ki≤N)。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如：3,3,1,2,5代表了Ki(K1=3,K2=3,…)，从1楼开始。在1楼，按“上”可以到4楼，按“下”是不起作用的，因为没有−2楼。那么，从A楼到B楼至少要按几次按钮呢？

输入

共二行。
第一行为3个用空格隔开的正整数，表示N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为N个用空格隔开的非负整数，表示Ki。

输出

输出一行，即最少按键次数,若无法到达，则输出−1。

样例输入

5 1 5
3 3 1 2 5

样例输出

3

分析

使用BFS算法。很有趣的一道题，“我做过一个梦”，哈哈 。

#include <bits/stdc++.h>

using namespace std;

struct node{
    int now;//当前位于多少层
    int level;//当前走了几次
};
int main()
{
    int n,org,des,ans = 0;
    cin>>n>>org>>des;
    int elv[n];

    bool visited[n];
    queue<node> que;
    memset(visited,false,sizeof(visited));
    for(int i = 0;i<n;i++)
        cin>>elv[i];
    node tmp;
    tmp.now = org - 1;
    tmp.level = 0;
    que.push(tmp);
    visited[org - 1] = true;
    while(!que.empty()){
        node no = que.front();
        que.pop();
        if(no.now == des - 1){
            cout<<no.level;
            return 0;
        }

        if(no.now + elv[no.now] < n&&visited[no.now+elv[no.now]] == false){
            //判断下一个点的可行性及是否到过
            visited[no.now+elv[no.now]] = true;
            tmp.now = no.now+elv[no.now];
            tmp.level = no.level+1;
            que.push(tmp);
        }
        if(no.now - elv[no.now] >= 0&&visited[no.now+elv[no.now]] == false){
            visited[no.now-elv[no.now]] = true;
            tmp.now = no.now-elv[no.now];
            tmp.level = no.level+1;
            que.push(tmp);
        }
    }
    cout<<-1;
    return 0;
}