开始学图论……先从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;
}