neuq-acm预备队训练week 6 P1135 奇怪的电梯

题目描述

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

题目限制

输入格式

共二行。

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

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

输出格式

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

输入输出样例

解题思路

本题为一个简单的BFS问题，主要思路看下方代码

AC代码

#include <bits/stdc++.h>
using namespace std;
int n,s,e;
int a[201],v[201];

struct pos
{
    int level;
    int steps;
};
void bfs();

int main()
{
    cin>>n>>s>>e;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    memset(v,0,sizeof(v));
    bfs();

    return 0;
}

void bfs()
{
    pos cur,nex;
    cur.level=s;
    cur.steps=0;
    queue<pos>q;
    q.push(cur);   //将当前起始第一个入队列
    v[s]=1;        //标记
    while(!q.empty())   //队列非空
    {
        cur=q.front();  //令cur为队首
        q.pop();        //上一步已取得，现将队首pop出去，保证后面入队的为新队首
        if(cur.level==e)  //若到目标层，结束
        {
            printf("%d",cur.steps);
            return ;
        }

        nex.level=cur.level+a[cur.level];   //向上走
        nex.steps=cur.steps+1;
        if(nex.level<=n)    //楼层合法
        {
            if(v[nex.level]==0)    //未访问
            {
                v[nex.level]=1;
                q.push(nex);      //入队列
            }
        }

        nex.level=cur.level-a[cur.level];  //向下走
        nex.steps=cur.steps+1;
        if(nex.level>=1)    //楼层合法
        {
            if(v[nex.level]==0)    //未访问
            {
                v[nex.level]=1;
                q.push(nex);    //入队列
            }
        }

    }
    printf("-1");   //找不到
}