洛谷 P1135 奇怪的电梯

//int队列存储层数的bfs
#include<iostream>
#include<queue>
using namespace std;
const int N = 310;

int n, A, B;
queue<int>q;
int step;
bool vis[N];
int a[N];

int bfs() {
    q.push(A);
    vis[A] = true;
    while (++step) {				//每次对所有对头枚举过后，也就是要走下一步的时候，把走过的步数加一
        if (q.empty()) {	//如果队列已经空了，就说明无法到达目标楼层
            return -1;
        }
        int len = q.size();	//在此处先行存储下当前的队列长度，因为下面循环里会使队列长度变化

        for (int i = 0; i < len; i++) {		//循环遍历取出上一次压入的所有层数（不包含在枚举过程中压入的下一次枚举的层数）

            int t = q.front();
            q.pop();

	    //枚举出对头往上走和往下走的情况
            if (t + a[t] <= n) {		//如果往上走是合法的
                if (!vis[t + a[t]]) {		//如果这个状态没有经历过
                    if (t + a[t] == B) {	//如果现在已经到达了目标层数
                        return step;
                    }
                    vis[t + a[t]] = true;	//标记上此状态
                    q.push(t + a[t]);		//把这一层数压入队列，以进行下一次的层数枚举
                }
            }

            if (t - a[t] >= 1) {		//如果往下走是合法的
                if (!vis[t - a[t]]) {
                    if (t - a[t] == B) {
                        return step;
                    }
                    vis[t - a[t]] = true;
                    q.push(t - a[t]);
                }
            }
        }
    }
    return -1;
}

int main() {
    cin >> n >> A >> B;

    if (A == B) {		//如果从一开始，起始楼层就等于目标楼层，那么直接输出0步
        printf("0");
        return 0;
    }

    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }

    int ans = bfs();
    printf("%d", ans);

    return 0;
}