题目链接: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();
}
}