【bfs】奇怪的电梯（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

分析

bfs，裸广搜

参考代码

 

///bfs
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
int N,A,B;
int a[201],visited[201];
int ans[201];
int flag=0;
queue <int> q;

void bfs(int ceng);
int main()
{
    scanf("%d%d%d",&N,&A,&B);
    for(int i=1;i<=N;i++){
        scanf("%d",&a[i]);
    }
    q.push(A);
    visited[A]=1;
    ans[A]=0;
    bfs(1);
    if(flag==0)
        printf("-1\n");
    return 0;
}
void bfs(int ceng){
    while(!q.empty()){
        int t=q.front();
        if(t==B){
            flag=1;
            printf("%d\n",ans[B]);
            break;
        }
        int tt;
        q.pop();
        for(int i=0;i<=1;i++){
            if(i==0){
                tt=t+a[t];
                if(!visited[tt]&&tt<=N){
                    q.push(tt);
                    visited[tt]=1;
                    ans[tt]=ans[t]+1;
                }
            }
            else if(i==1){
                tt=t-a[t];
                if(!visited[tt]&&tt>=1){
                    q.push(tt);
                    visited[tt]=1;
                    ans[tt]=ans[t]+1;
                }
            }
        }
    }

}