bfs——奇怪的电梯

奇怪的电梯

题目背景

感谢 @yummy 提供的一些数据。

题目描述

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

输入格式

共二行。

第一行为三个用空格隔开的正整数，表示 $N,A,B$（$1\le N\le 200$，$1\le A,B\le N$）。

第二行为 $N$ 个用空格隔开的非负整数，表示 $K_i$。

输出格式

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

样例 #1

样例输入 #1

5 1 5
3 3 1 2 5

样例输出 #1

3

提示

对于 $100\%$ 的数据，$1\le N\le 200$，$1\le A,B\le N$，$0\le K_i\le N$。

本题共 $16$ 个测试点，前 $15$ 个每个测试点 $6$ 分，最后一个测试点 $10$ 分。

思路

我们就直接建图，看每个楼层能到达的楼层数，如果超过总楼层数则不要，最后我们再用bfs来求最短路即可（因为本道题边权为1）。

代码

//最短路问题，某一层楼能上/下的层数取决于k

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;

const int N = 210,M = 2*N;

int h[N],e[M],ne[M],w[M],idx;
int dist[N];
bool st[N];
queue<int> q;
int n,a,b;

void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void bfs(){
    q.push(a);
    
    st[a]=true;
    dist[a]=0;
    
    while(q.size()){
        int t=q.front();
        q.pop();
        
        for(int i=h[t];~i;i=ne[i]){
            int j=e[i];
            if(st[j])continue;
            if(dist[j]>dist[t]+1){
                dist[j]=dist[t]+1;
                st[j]=true;
                q.push(j);
            }
        }
    }
}

int main(){
    cin>>n>>a>>b;
    
    memset(h,-1,sizeof h);
    
    memset(dist,0x3f,sizeof dist);
    
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        if(i-x>=1)add(i,i-x);
        if(i+x<=n)add(i,x+i);
    }
    
    bfs();
    
    cout<<(dist[b]==0x3f3f3f3f?-1:dist[b]);
    return 0;
}