HDU 1548 A strange lift

开始学图论……先从dijkstra开始……写点东西供以后翻阅……转载请注明该文链接

题意：有一个电梯，每一层有一个数字k，并有两种选择：选UP会让电梯向上k层，选DOWN会让电梯向下k层。给起点a终点b，问从a到b最少步数。

思路：每一层记为一个结点，第i层能到达i-k和i+k层，则在i->i-k与i->i+k各建一条边（前提i-k>=1、i+k<=n），然后跑一遍dijkstra或者spfa，可以直接出来结果。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>

using namespace std;

const int maxn=210;
const int inf=0x3f3f3f3f;

int E[maxn][maxn],n;
int p[maxn],d[maxn];

void Dijkstra(int s)
{
    bool vis[maxn];
    memset(vis,false,sizeof(vis));
    memset(p,0,sizeof(p));
    memset(d,inf,sizeof(d));
    d[s]=0;
    while(1)
    {
        int u=-1;
        for(int i=0;i<n;i++)
        {
            if(!vis[i] && (u==-1 || d[i]<d[u])) u=i;
        }
        if(u==-1 || d[u]==inf) break;
        vis[u]=true;
        for(int v=0;v<n;v++)
        {
            if(d[u]+E[u][v]<d[v])
            {
                d[v]=d[u]+E[u][v];
                p[v]=u;
            }
        }
    }
}

int main()
{
    int a,b,m;
    while(scanf("%d",&n),n)
    {
        memset(E,inf,sizeof(E));
        scanf("%d%d",&a,&b);
        a--;
        b--;
        for(int i=0;i<n;i++)
        {
            scanf("%d",&m);
            if(i-m>=0) E[i][i-m]=1;
            if(i+m<n ) E[i][i+m]=1;
        }
        Dijkstra(a);
        if(d[b]!=inf) printf("%d\n",d[b]);
        else printf("-1\n");
    }
    return 0;
}