poj1548 A strange lift

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1548

题意：有一个奇怪的电梯，在每一层上只能上升或下降特定的层数，比如，在第i层给的数字是k，则在这层可以下降到i-k，或者上升到i+k，但是楼层必须在1-n之间。输入，第一行三个数，没别是楼层，以及起点和终点，第二层n个数，表示每个楼层能上升或下降的楼层数。

想法：先构图，有向图，如果输入时能够到达某些楼层则让他们之间的数标记为1，然后就是求最短路了。。具体看代码。

代码；

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=210;
bool vis[maxn];
int link[maxn][maxn],dis[maxn];
int n,a,b;
queue <int> ss;
void spfa(){
    for(int i=1;i<=n;i++)
        dis[i]=INF;
    dis[a]=0;
    memset(vis,false,sizeof(vis));
    vis[a]=true;
    ss.push(a);
    while(!ss.empty()){
        int x=ss.front();
        ss.pop();
        vis[x]=false;
        for(int i=1;i<=n;i++){
            if(dis[i]>dis[x]+link[x][i]){
                dis[i]=dis[x]+link[x][i];
                if(!vis[i]){
                    vis[i]=true;
                    ss.push(i);
                }
            }
        }
    }
    if(dis[b]==INF)  cout<<"-1"<<endl;
    else  cout<<dis[b]<<endl;
}
int main(){
    while(~scanf("%d",&n)&&n){
        scanf("%d%d",&a,&b);
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                link[i][j]=INF;
            }
        }
        for(int i=1;i<=n;i++){
            int k;
            scanf("%d",&k);
            if(i+k<=n)
                link[i][i+k]=1;
            if(i-k>0)
                link[i][i-k]=1;
        }
        spfa();
    }
    return 0;
}