奇怪的电梯（BFS）

慧狗仔

于 2024-04-28 09:00:00 发布

阅读量1k

点赞数 32

文章标签：宽度优先算法深度优先 c++

本文链接：https://blog.csdn.net/2301_80065123/article/details/138251973

版权

奇怪的电梯（BFS）

题目背景

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

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 $i$ 层楼（ $\le i \le N$ ）上有一个数字 $K_i$ （ $\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$ （ $\le N \le 200$ ， $\le A, B \le N$ ）。

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

输出格式

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

样例 #1

样例输入 #1

5 1 5
3 3 1 2 5

样例输出 #1

提示说明

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

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

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<ll,ll> PII;
const ll N=210;

ll k[N];// n 层楼(3,3,1,2,5)
ll vis[N];//标记
ll n,a,b;// n 层楼(A-->B)
queue <PII> q;//优先队列

ll bfs()
{
    //将vis数组所有元素初始化为 0
    memset(vis,0,sizeof(vis));
    //将第 A 层楼, 0 次入队
    q.push({a,0});

    //当队列非空时执行循环
    while(q.size())
    {
        auto t=q.front();
        q.pop();

        // v 楼层， p 上下的次数
        ll v=t.first,p=t.second;

        //终止搜索，返回最短次数 p
        if(v==b) return p;

        //上，下
        ll up=v+k[v],of=v-k[v];
        if(up<=n&&!vis[up])//上
        {
            //进行下一次入队
            q.push({up,p+1});
            vis[up]=1;//标记
        }
        if(of>=1&&!vis[of])//下
        {
            //进行下一次入队
            q.push({of,p+1});
            vis[of]=1;//标记
        }
    }
    return -1;
}

int main()
{
    cin>>n>>a>>b;

    for(ll i=1;i<=n;i++)
        cin>>k[i];

    ll res=bfs();
    cout<<res<<endl;
    return 0;
}