EOJ2124 Seamild的电梯

http://acm.cs.ecnu.edu.cn/problem.php?problemid=2124

题意，电梯第i层能上升Ki层或下降Ki层，问从a到b的最短操作次数

最短路floyd，建图时即是把x到x+ki x-ki (若存在)赋值为1，开始存为inf。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int mat[105][105];
const int inf=99999999;
void floyd(int n){
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(mat[i][k]+mat[k][j] < mat[i][j])
                    mat[i][j]=mat[i][k]+mat[k][j];
}
int main(){
    int n,a,b;
    while(scanf("%d%d%d",&n,&a,&b)!=EOF){
        
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                mat[i][j]=inf;
        for(int i=1;i<=n;i++){
            int x;
            scanf("%d",&x);
            if(i-x>0) mat[i][i-x]=1;          //建图
            if(i+x<=n) mat[i][i+x]=1;
        }
        if(a==b){cout<<"0"<<endl;continue;}  //特判一下
        floyd(n);
        if(mat[a][b]<inf)cout<<mat[a][b]<<endl; 
        else cout<<"-1"<<endl;                   //不连通，无法到达
    }
    return 0;
}