P1135 奇怪的电梯

 复制Markdown  展开

题目背景

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

题目描述

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

输入格式

共二行。

第一行为三个用空格隔开的正整数，表示 �,�,�N,A,B（1≤�≤2001≤N≤200，1≤�,�≤�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;
struct floors{
    int up,down,step;
    bool visit;
}fs[205];
int main(){
    int n,a,b,tmp;
    cin>>n>>a>>b;
    for(int i=1;i<=n;i++){
        cin>>tmp;
    fs[i].up=i+tmp;
    fs[i].down=i-tmp;
    fs[i].step=-1;
    fs[i].visit=false;
    }
    queue<floors> q;
    fs[a].step=0,fs[a].visit=true;
    q.push(fs[a]);
    while(!q.empty()){
        floors f=q.front();
        q.pop();
        int u=f.up,d=f.down;
        if(u>=1&&u<=n&&fs[u].visit==false){
            fs[u].visit=true;
            fs[u].step=f.step+1;
            q.push(fs[u]);
        }
        if(d>=1&&d<=n&&fs[d].visit==false){
            fs[d].visit=true;
            fs[d].step=f.step+1;
            q.push(fs[d]);
        }
        if(u==b||d==b) break;
    }
    cout<<fs[b].step;
    return 0;
}