hdu 1548 A strange lift

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int inf=10024;
int mp[200+5][200+5],dist[200+5],n,vis[200+5];

void Dijkstra(int v0)
{
    int i,j,k,u,minx;
    for(i=1;i<=n;i++)
    {
        dist[i]=mp[v0][i];
        vis[i]=0;
    }
    vis[v0]=1;
    dist[v0]=0;
    for(i=0;i<n-1;i++)
    {
        minx=inf,u=v0;
        for(j=1;j<=n;j++)
        {
            if(!vis[j]&&dist[j]<minx)
            {
                minx=dist[j];
                u=j;
            }
        }
        vis[u]=1;
        for(k=1;k<=n;k++)
        {
            if(!vis[k]&&mp[u][k]<inf)
                dist[k]=min(dist[k],dist[u]+mp[u][k]);
        }
    }
}

int main()
{
    int a,b,k,i,j,x,y;
    while(~scanf("%d",&n)&&n)
    {
        scanf("%d%d",&a,&b);
        memset(mp,0,sizeof(mp));
        for(i=1; i<=n; i++)
        {
            scanf("%d",&x);
            if(i-x>=1) mp[i][i-x]=1;
            if(x+i<=n) mp[i][x+i]=1;
        }
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=n;j++)
            {
                if(i==j) mp[i][j]=0;
                else if(mp[i][j]==0) mp[i][j]=inf;
            }
        }
        Dijkstra(a);
        if(a==b) printf("0\n");
        else if(dist[b]<inf) printf("%d\n",dist[b]);
        else printf("-1\n");
    }
    return 0;
}

你可能感兴趣的:(hdu 1548 A strange lift)