hdu 1548 最短路

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1548

就是Dijkstra算法的应用。。。。不过被坑了好久，原来电梯的up、down是并列的，初始化的时候出错了。。。。调了好半天。。突然醒悟过来了，，，orz

#include<iostream>
#include<cstring>
const int N=220;
const int inf=1000000;
using namespace std;

int edge[N][N];
int dist[N];
int visited[N];
int n,A,B;

/*
void floyd(){
    for(int k=1;k<=n;k++){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(edge[i][k]+edge[k][j]<edge[i][j]){
                    edge[i][j]=edge[i][k]+edge[k][j];
                    path[i][j]=path[i][k];
                }
            }
        }
    }
    if(edge[A][B]==inf){
        printf("-1\n");
    }else {
        int count=1;
        int k=A;
        while(path[k][B]!=B){
            k=path[k][B];
            count++;
        }
        printf("%d\n",count);
    }
}*/

void Dijkstra(int v0){
    memset(visited,0,sizeof(visited));
    for(int i=1;i<=n;i++){
        dist[i]=edge[v0][i];
    }
    dist[v0]=0;
    visited[v0]=1;
    for(int i=1;i<n;i++){
        int min=inf,u=v0;
        for(int j=1;j<=n;j++){
            if(!visited[j]&&dist[j]<min){
                u=j,min=dist[j];
            }
        }
        if(min==inf)return ;
        visited[u]=1;
        for(int k=1;k<=n;k++){
            if(!visited[k]&&edge[u][k]<inf&&edge[u][k]+dist[u]<dist[k]){
                dist[k]=dist[u]+edge[u][k];
            }
        }
    }
}

int main(){
    while(scanf("%d",&n)!=EOF){
        if(n==0)break;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
            //    path[i][j]=j;//此时为顺序输出
                if(i==j){
                    edge[i][j]=0;
                }else 
                    edge[i][j]=inf;
            }
        }
        int num;
        scanf("%d%d",&A,&B);
        for(int i=1;i<=n;i++){
            scanf("%d",&num);
            if(i+num<=n)edge[i][i+num]=1;//两个状态时并列的，orz。。。
            if(i-num>=1)edge[i][i-num]=1;//
        }
        //floyd();
        Dijkstra(A);
        if(dist[B]<inf){
            printf("%d\n",dist[B]);
        }else 
            printf("-1\n");
    }
    return 0;
}