洛谷 P1135 奇怪的电梯

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第i层楼(1<=i<=N)上有一个数字Ki(0<=Ki<=N)。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如：3 3 1 2 5代表了Ki(K1=3,K2=3,……)，从一楼开始。在一楼，按“上”可以到4楼，按“下”是不起作用的，因为没有-2楼。那么，从A楼到B楼至少要按几次按钮呢？

输入输出格式

输入格式：

输入文件共有二行，第一行为三个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N)，第二行为N个用空格隔开的正整数，表示Ki。

输出格式：

输出文件仅一行，即最少按键次数,若无法到达，则输出-1。

输入输出样例

输入样例#1：

LIFT.IN
5 1 5
3 3 1 2 5

输出样例#1：

LIFT.OUT
3

 

 

最短路

屠龙宝刀点击就送

#include <cstring>
#include <cstdio>
#include <queue>
#define N 2005
using namespace std;
bool vis[N];
int n,a,b,cnt,nextt[N],to[N],head[N],dis[N];
struct node
{
    int x,y;
    bool operator<(node a)const
    {
        return y>a.y;
    }
};
priority_queue<node>q;
inline void ins(int u,int v)
{
    nextt[++cnt]=head[u];
    to[cnt]=v;
    head[u]=cnt;
}
int Main()
{
    scanf("%d%d%d",&n,&a,&b);
    for(int k,i=1;i<=n;++i)
    {
        scanf("%d",&k); 
        if(i-k>0) ins(i,i-k);
        if(i+k<=n) ins(i,i+k);
    }
    for(int i=1;i<=n;++i) dis[i]=0x3f3f3f3f;
    dis[a]=0;
    q.push((node){a,dis[a]});
    for(node tmp;!q.empty();)
    {
        tmp=q.top();
        q.pop();
        if(vis[tmp.x]) continue;
        vis[tmp.x]=1;
        for(int i=head[tmp.x];i;i=nextt[i])
        {
            int v=to[i];
            if(dis[v]>dis[tmp.x]+1)
            {
                dis[v]=dis[tmp.x]+1;
                q.push((node){v,dis[v]}); 
            }
        }
    }
    dis[b]!=0x3f3f3f3f?printf("%d
",dis[b]):printf("-1");
    return 0;
}
int sb=Main();
int main(int argc,char *argv[]){;}