奇怪的電梯(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 = 1<<30;
 
int n;
int map[205][205];
int a[205],cnt;
int vis[205],cast[205];
 
void Dijkstra(int s,int e)
{
    int i,j,min,pos;
    memset(vis,0,sizeof(vis));
    for(i = 0; i<n; i++)
        cast[i] = map[s][i];
    cast[s] = 0;
    vis[s] = 1;
    for(i = 1; i<n; i++)
    {
        min = inf;
        for(j = 0; j<n; j++)
        {
            if(cast[j]<min && !vis[j])
            {
                pos = j;
                min = cast[j];
            }
        }
        if(min == inf)
            break;
        vis[pos] = 1;
        for(j = 0; 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 = 0; i<n; i++)
            for(j = 0; j<n; j++)
                map[i][j] = inf;
        for(i = 0; i<n; i++)
        {
            scanf("%d",&a[i]);
            if(i+a[i]<n)
                map[i][i+a[i]] = 1;
            if(i-a[i]>=0)
                map[i][i-a[i]] = 1;
        }
        Dijkstra(s,e);
        printf("%d\n",cast[e]==inf?-1:cast[e]);
    return 0;
}

后來用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+10;
using namespace std;
int n,m,a,b;
const int dis[][2]={0,1,1,0,-1,0,0,-1};
int q[maxn],vis[maxn];
int main()
{
    int i,j,k,t;
    queue<int>p;
    scanf("%d%d%d",&n,&a,&b);
    vis[a]=1;
    for(i=1;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>=1&&!vis[l])p.push(l),vis[l]=vis[now]+1;
        if(r<=n&&!vis[r])p.push(r),vis[r]=vis[now]+1;
        if(l==b||r==b)break;
    }
    printf("%d\n",vis[b]-1);
    //system("pause");
    return 0;
}