奇怪的电梯(HDU1548) （Dijkstra）或者（BFS）

问题 E: 奇怪的电梯

时间限制: 1 Sec  内存限制: 64 MB
提交: 35  解决: 16
[提交][状态][讨论版]

题目描述

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

输入

第1行为三个正整数，表示N，A，B(1≤N≤200，1≤A，B≤N)；
第2行为N个正整数，表示Ki。

输出

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

样例输入

5 1 5
3 3 1 2 5

样例输出

3

先试试Dijkstra，发现过了，代码如下

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

const int inf = <<;

int n;
int map[][];
int a[],cnt;
int vis[],cast[];

void Dijkstra(int s,int e)
{
    int i,j,min,pos;
    memset(vis,,sizeof(vis));
    for(i = ; i<n; i++)
        cast[i] = map[s][i];
    cast[s] = ;
    vis[s] = ;
    for(i = ; i<n; i++)
    {
        min = inf;
        for(j = ; j<n; j++)
        {
            if(cast[j]<min && !vis[j])
            {
                pos = j;
                min = cast[j];
            }
        }
        if(min == inf)
            break;
        vis[pos] = ;
        for(j = ; j<n; j++)
        {
            if(cast[pos]+map[pos][j]<cast[j] && !vis[j])
                cast[j] = cast[pos]+map[pos][j];
        }
    }
}

int main()
{
    int i,j,s,e,x,y;
    scanf("%d",&n);
      scanf("%d%d",&s,&e);
        s--,e--;
        for(i = ; i<n; i++)
            for(j = ; j<n; j++)
                map[i][j] = inf;
        for(i = ; i<n; i++)
        {
            scanf("%d",&a[i]);
            if(i+a[i]<n)
                map[i][i+a[i]] = ;
            if(i-a[i]>=)
                map[i][i-a[i]] = ;
        }
        Dijkstra(s,e);
        printf("%d\n",cast[e]==inf?-:cast[e]);
    return ;
}

后来用BFS交了一发，也过了。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<map>
#include <queue>
#define ll long long
#define rep(i,m,n) for(i=m;i<=n;i++)
#define inf 0x3f3f3f3f
const int maxn=1e3+;
using namespace std;
int n,m,a,b;
const int dis[][]={,,,,-,,,-};
int q[maxn],vis[maxn];
int main()
{
    int i,j,k,t;
    queue<int>p;
    scanf("%d%d%d",&n,&a,&b);
    vis[a]=;
    for(i=;i<=n;i++)scanf("%d",&q[i]);
    p.push(a);
    while(!p.empty())
    {
        int now=p.front(),l=now-q[now],r=now+q[now];
        p.pop();
        if(l>=&&!vis[l])p.push(l),vis[l]=vis[now]+;
        if(r<=n&&!vis[r])p.push(r),vis[r]=vis[now]+;
        if(l==b||r==b)break;
    }
    printf("%d\n",vis[b]-);
    //system("pause");
    return ;
}