hdu 1548 A strange lift (dijkstra算法)

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

题目大意：升降电梯，先给出n层楼，然后给出起始的位置，即使输出从A楼道B楼的最短时间。

注意的几点

(1)每次按一下，只能表示上或者是下，然后根据输入的看是上几层或者是下几层。

(2)注意不能到底不存在的楼层。

详见代码。

#include <iostream>
#include <cstdio>
using namespace std;
const int INF=9999999;
int map[205][205],node[205],vis[205],Min,n;

void set()
{
    for (int i=1; i<=n; i++)
    {
        vis[i]=0;
        node[i]=INF;
        for (int j=1; j<=n; j++)
            map[i][j]=INF;
    }
}

int dijkstra(int m,int end)
{
    int tm=m;
    vis[m]=1;
    node[m]=0;
    for (int k=2; k<=n; k++)
    {
        Min=INF;
        for (int i=1; i<=n; i++)
            if (!vis[i])
            {
                if (node[i]>node[tm]+map[tm][i])
                    node[i]=node[tm]+map[tm][i];
                if (Min>node[i])
                {
                    Min=node[i];
                    m=i;
                }
            }
        vis[m]=1;
        tm=m;
        if(vis[end])
            return node[end];
    }
    return -1;
}

int main ()
{
    int k;
    while (scanf("%d",&n),n)
    {
        set();
        int a,b;
        scanf("%d%d",&a,&b);
        for (int i=1; i<=n; i++)
        {
            scanf("%d",&k);
            if (i+k<=n)
                map[i][i+k]=1;
            if (i-k>=1)
                map[i][i-k]=1;
        }
        printf ("%d
",dijkstra(a,b));
    }
    return 0;
}