杭电 1548 A strange lift (Dijkstra)

传送门：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;
}