HDU 1548 A strange lift（bfs或dijkstra）

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

题意：每层电梯对应着一个一个数k[i](i为这层电梯的层号），你可以选择UP或者DOWN，若选择UP则上升k[i]层，若现在DOWN则下降k[i]层，需要注意的是下降或上升之后的层数需在1-n之内。

如何用dijkstra实现：

若i层楼梯可以上升或者可以下降到某层，则设置i层电梯到达那一层的距离为1，然后在用dijkstra得到a->b的最短距离就行了。

代码：

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=500;
int vis[maxn],dis[maxn],mp[maxn][maxn],n;
void djk(int v){//最普通的dijkstra
    int i,j,k;
    memset(dis,inf,sizeof(dis));
    for(i=1;i<=n;i++)
    dis[i]=mp[v][i];
    for(i=0;i<n-1;i++){
        int mi=inf,u;
        for(j=1;j<=n;j++){
            if(dis[j]<mi&&!vis[j]){
                mi=dis[j];
                u=j;
            }
        }
        vis[u]=1;
        for(k=1;k<=n;k++){
            if(dis[k]>dis[u]+mp[u][k]){
                dis[k]=dis[u]+mp[u][k];
            }
        }
    }
}
int main(){
    int a,b,i,j;
    while(cin>>n&&n){
        cin>>a>>b;
        int k[maxn];
        memset(mp,inf,sizeof(mp));
        memset(vis,0,sizeof(vis));
        for(i=1;i<=n;i++){
            cin>>k[i];
        }
        for(i=1;i<=n;i++){//若i层楼梯上升或下降k[i]层后依然属于1~n，则令其距离等于1
            mp[i][i]=0;//i层到自己的距离为0
            if(i-k[i]>0)
            mp[i][i-k[i]]=1;
            if(i+k[i]<=n)
            mp[i][i+k[i]]=1;
        }
        djk(a);
        if(dis[b]<inf)
        cout<<dis[b]<<endl;
        else
        cout<<-1<<endl;
    }
}

如何用bfs实现：

一种非常便捷的选择是用临界表将图存下来，然后进行bfs，不过你也可以选择不用邻接表存图，反正i层电梯可以直接到达的电梯数顶多两个。

代码：

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=205;
int vis[maxn],n;
vector<int>link[maxn];
struct node{
    int p,t;
};
int bfs(int v,int l){
    queue<node>q;
    int i,j,k;
    node u,r;
    u.p=v;u.t=0;
    q.push(u);
    vis[u.p]=1;
    while(!q.empty()){
        u=q.front();
        q.pop();
        if(u.p==l){//若到达l(即b层电梯）返回
            return u.t;
        }
            for(i=1;i<=link[u.p][0];i++){//遍历u.p层电梯可以到达的电梯层，不超过2，哈哈哈哈哈哈
                r.p=link[u.p][i];
                r.t=u.t+1;
                if(!vis[r.p]){
                vis[r.p]=1;
                q.push(r);}
            }
    }
    return -1;//若无法到达则返回-1
}
int main(){
    int a,b,i,j,k; 
    while(cin>>n&&n){
        int k[maxn];
        memset(vis,0,sizeof(vis));
        cin>>a>>b;
        for(i=1;i<=n;i++)
        cin>>k[i];
        for(i=1;i<=n;i++)
        link[i].push_back(0);
        for(i=1;i<=n;i++){//邻接表存图
            if(i-k[i]>0){
                link[i][0]++;//link[i][0]代表i层电梯可以到达的电梯的数量
                link[i].push_back(i-k[i]);
            }
            if(i+k[i]<=n){
                link[i][0]++;
                link[i].push_back(i+k[i]);
            }
        }
        cout<<bfs(a,b)<<endl;
        for(i=1;i<=n;i++)//清空，此处是必要的
        link[i].clear();
    }
}