专题二 · 1013

代码及解释

#include <iostream>
#include <cstring>
#include <queue>

// 有一天我坐了一个奇怪的电梯
// 这个电梯上下的时候会按照他的心情来走＝－＝
// 然后问我用最少的路数到某一层
// （想到了文理大楼老是夹人的电梯……

// 很标准的广搜，寻找最优可行解

int floor[1000];
bool vis[1000];

struct
Info { int k; int pos; };

int bfs(int start, int end, int floor_num) {
    vis[start] = true;
    std::queue <Info> queue;
    
    Info info = {start, 0};
    
    queue.push(info);
    
    
    while(!queue.empty()) {
        Info t = queue.front();
        queue.pop();
        
        // 找到辣
        if(t.k == end) return t.pos;
        
        Info tmp;
        
        // 循环加入队列
        for(int i = -1; i <= 1; i++) {
            tmp.k = t.k + i * floor[t.k];
            
            if(tmp.k >= 1 && tmp.k <= floor_num && !vis[tmp.k]) {
                vis[tmp.k] = true;
                tmp.pos = t.pos + 1;
                queue.push(tmp);
            }
        }
        
    }
    return -1;
}


int main() {
    using std::cin;
    using std::cout;
    using std::endl;
    
    int start, end, floor_num;
    
    while(cin >> floor_num) {
        if(!floor_num) break;
        cin >> start >> end;
        
        memset(vis, 0, sizeof(vis));
        
        for(int i = 1; i <= floor_num; i++)
            cin >> floor[i];
        
        int pos = bfs(start, end, floor_num);
        cout << pos << endl;
    }
    
    return 0;
}