青藤 #10064 奇怪的电梯

题目描述

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

思路

直接 BFS ，具体看注释。

#include <iostream>
#include <queue>
using namespace std;
int spoint, fpoint, lift[100001], n, ud[3] = { 0, 1, -1 }, click;
int isper[100001];
queue<int> route;
int BfsLift() {
    int want;
    if (spoint == fpoint)
        return 0;
    route.push(spoint);
    isper[spoint] = 0;
    while (!route.empty()) {
        for (int i = 1; i <= 2; i++) {
            want = ud[i] * lift[route.front()] + route.front();//将要到的电梯楼层
            if (want >= 1 && want <= n && isper[want] == 0) {//没有出电梯楼层&&当前楼层最优解
                isper[want] = isper[route.front()] + 1;//按下一次电梯
                route.push(want);//进入队尾
            }
            if (want == fpoint)//有解
                return isper[want];
        }
        route.pop();
    }
    return -1;
}
int main() {
    cin >> n >> spoint >> fpoint;
    for (int i = 1; i <= n; i++) cin >> lift[i];
    cout << BfsLift();
}

注意事项：不要用单独的量来统计，否则无法达到最优解。