奇怪的电梯

题目描述

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

输入

第1行为3个用1个空格隔开的正整数，表示N、A、B(1<=N<=2001<=AB<=N)。

第2行为N个用空格隔开的正整数，表示Ki。

输出

一行一个数，表示最少按键次数。若无法到达，则输出“-1”。

样例输入

5 1 5
3 3 1 2 5

样例输出

3

思路

又是一道很水的搜索，在深搜里判断一下电梯是往上走还是往下走就行了，可以当模板。

代码

#include<iostream>
using namespace std;
int n,x,y,minn=999999;
int a[201],book[201];
void dfs(int m,int cnt) {
    if(m==y) {//达到目标. 
        minn=min(minn,cnt);
    }
    else if(cnt<=minn) {//简单的剪枝 
            book[m]=1;
            if(m+a[m]<=n && book[m+a[m]]==0) dfs(m+a[m],cnt+1);//上升 
            if(m-a[m]>=1 && book[m-a[m]]==0) dfs(m-a[m],cnt+1);//下降 
            book[m]=0;//回溯 
    }
}
int main() {
    cin>>n>>x>>y;
    for(int i=1; i<=n; i++) {
        cin>>a[i];
    }
    dfs(x,0);
    if(minn==999999) cout<<-1;//到达不了 
    else cout<<minn;
    return 0;
}