hdu 1548：A strange lift

一部电梯，能上能下。在第i层有一个k，电梯在i层只能上到第i + k或下到第i – k层，且不会超过第N层或低于第1层。问从A层到B层最少上升或下降几次。若无法到达则输出-1。

 

Spfa简单搞定。不过这题既视感好强。。。

#include <cstdio>
#include <cstring>
#include <queue>
#include <iostream>
using namespace std;

const int inf = 1 << 20 ;
const int MAX = 205 ;
int N , A , B ;
int kk[MAX] ;
int dis[MAX] ;
bool had[MAX] ;
queue <int> q ;

void spfa() {
    for (int i = 1 ; i <= N ; i ++) {
        dis[i] = inf ;
    }
    dis[A] = 0 ;
    q.push(A) ; had[A] = 1 ;
    while (!q.empty()) {
        int t = q.front() ; q.pop() ;
        had[t] = 0 ;
        int t_p = t + kk[t] ;
        int t_m = t - kk[t] ;
        if (t_p <= N && dis[t_p] > dis[t] + 1) {
            dis[t_p] = dis[t] + 1 ;
            if (!had[t_p]) {
                had[t_p] = 1 ;
                q.push(t_p) ;
            }
        }
        if (t_m >= 1 && dis[t_m] > dis[t] + 1) {
            dis[t_m] = dis[t] + 1 ;
            if (!had[t_m]) {
                had[t_m] = 1 ;
                q.push(t_m) ;
            }
        }
    }
}

int main() {
    while (cin >> N) {
        if (!N) break ;
        cin >> A >> B ;
        for (int i = 1 ; i <= N ; i ++) {
            cin >> kk[i] ;
        }
        spfa() ;
        if (dis[B] == inf) cout << -1 << endl ;
        else cout << dis[B] << endl ;
    }
    return 0 ;
}