【洛谷 P1135】奇怪的电梯题解（广度优先搜索）-CSDN博客呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。0≤

奇怪的电梯

题目描述

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

输入格式

共二行。

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

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

输出格式

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

样例 #1

样例输入 #1

5 1 5
3 3 1 2 5

样例输出 #1

提示

对于 100 % 100 \% 100% 的数据， 1 ≤ N ≤ 200 1 \le N \le 200 1≤N≤200， 1 ≤ A , B ≤ N 1 \le A, B \le N 1≤A,B≤N， 0 ≤ K i ≤ N 0 \le K_i \le N 0≤Ki≤N。

思路

搜到了就返回按按钮次数，搜不到返回-1。注意判断是否出界和重复访问。

AC代码

#include <iostream>
#include <queue>
#include <cstring>
#define AUTHOR "HEX9CF"
using namespace std;

const int maxn = 1005;

int k[1005];
int n, a, b;
queue<int> q;
bool vis[maxn];
int cnt[maxn];

int bfs(int a, int b){
    memset(vis, 0, sizeof(vis));
    memset(cnt, 0, sizeof(cnt));
    vis[a] = true;
    cnt[a] = 0;
    q.push(a);
    while(!q.empty()){
        int f = q.front();
        q.pop();
        // cout << f << endl;
        if(f == b){
            return cnt[b];
        }
        int t = f + k[f];
        if (t <= n && !vis[t])
        {
            cnt[t] = cnt[f] + 1;
            vis[t] = true;
            q.push(t);
        }
        t = f - k[f];
        if (t > 0 && !vis[t])
        {
            cnt[t] = cnt[f] + 1;
            vis[t] = true;
            q.push(t);
        }
    }
    return -1;
}

int main() {
    cin >> n >> a >> b;
    for(int i = 1; i <= n; i++){
          cin >> k[i];
    }
    cout << bfs(a, b) << endl;
    return 0;
}

【洛谷 P1135】奇怪的电梯 题解（广度优先搜索）-CSDN博客