传送门:http://acm.hdu.edu.cn/showproblem.php?pid=1548
题意:第一行给你电梯层数、你所在的层数及你想去的层数;第二行给你每层电梯的按钮K。这个电梯在i层只能往上走Ki层或往下走Ki层(不会到地下也不会上天),问你最少要按多少次按钮。
思路:最短路的Dijkstra算法,只要将互通两层之间的权值置为1就可以计算按键次。
代码:
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
#define INF 0x3f3f3f3f
int map[202][202];
bool vis[202];
int dis[202];
int n;
void Dijkstra(int s)
{
for(int i=1;i<=n;i++) dis[i]=map[s][i];
vis[s]=1;
dis[s]=0;
int k,tmp;
for(int i=1;i<=n;i++)
{
tmp=INF;
for(int j=1;j<=n;j++)
{
if(!vis[j]&&dis[j]<tmp)
{
tmp=dis[j];
k=j;
}
}
if(tmp==INF) break;
vis[k]=1; //记得置为一
for(int j=1;j<=n;j++)
{
if(!vis[j]) dis[j]=min(dis[j],dis[k]+map[k][j]);
}
}
}
int main()
{
int lop[202];
int a,b;
while(cin>>n&&n)
{
memset(map,INF,sizeof(map));
memset(dis,INF,sizeof(dis));
memset(vis,0,sizeof(vis));
scanf("%d%d",&a,&b);
for(int i=1;i<=n;i++)
{
scanf("%d",&lop[i]);
}
for(int i=1;i<=n;i++)//这个循环建立连通图
{
if(i-lop[i]>0) map[i][i-lop[i]]=1;
if(i+lop[i]<=n) map[i][i+lop[i]]=1;
}
Dijkstra(a);
if(dis[b]==INF) cout<<"-1\n";
else cout<<dis[b]<<endl;
}
return 0;
}