HDU1548 A strange lift

HDU1548 A strange lift

题意：
一个电梯，每一层都有up或者down，和一个ki代表上下的层数，给定起始和终止层数，求最少按up或者down的次数。

题解：
以层为顶点，所有层分别up或down到各自到达层为边（有向边），权重为1，建图。然后运用dijkstra求最短路即可。

AC代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair <int,int> P;
const int maxn = 200+10;
int N;
int A,B;
int a[maxn];
bool vis[maxn];

int bfs()
{
    queue <P> Q;
    Q.push(P(A,0));
    memset(vis,false,maxn);
    vis[A] = true;

    while(!Q.empty())
    {
        P t = Q.front();
        Q.pop();
        if(t.first == B)  return t.second;
        vis[t.first] = true;
        int up = t.first + a[t.first];
        int down = t.first - a[t.first];
        if(up <= N && !vis[up])   Q.push(P(up,t.second+1));
        if(down > 0&& !vis[down]) Q.push(P(down,t.second+1));
    }
   return -1;
}

int main()
{
  while(scanf("%d",&N) && N)
  {
      scanf("%d%d",&A,&B);
      for(int i = 1; i <= N; i++)
        scanf("%d",&a[i]);
      printf("%d\n",bfs());
  }
  return 0;
}