洛谷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： 复制
5 1 5
3 3 1 2 5
输出样例#1： 复制
3

直接放代码。bfs.

---

#include<iostream>
#include<cstdio>
using namespace std;

const int maxn=205;
const int zl[2]={1,-1};

int n,a,b,qwq[maxn],q1[maxn*2],q[maxn*2],flag[maxn];

void work()
{
    if(a==b) {
        printf("0");return ;
    }
    int k=1,head=0,tail=1,p=1;
    q[tail]=a;
    q1[tail]=0;
    flag[a]=1;
    while(head<tail){
        head++;
        for(int i=0;i<2;i++){//向上走或向下走
            if(!flag[zl[i]*qwq[q[head]]+q[head]]&&zl[i]*qwq[q[head]]+q[head]>=1&&zl[i]*qwq[q[head]]+q[head]<=n){
                q[++tail]=zl[i]*qwq[q[head]]+q[head];
                flag[zl[i]*qwq[q[head]]+q[head]]=1;
                q1[tail]=q1[head]+1;
            }
            if(q[tail]==b) {
                printf("%d",q1[tail]);
                return ;
            }
        }
    }
    printf("-1");return ;
}

int main()
{
    cin>>n>>a>>b;
    for(int i=1;i<=n;i++)
      cin>>qwq[i];
    work();
    return 0;
}

---