洛谷 P1135 奇怪的电梯 C++ 深搜

题目描述

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

输入格式

共二行。

第一行为3个用空格隔开的正整数，表示N,A,B(1≤N≤200,1≤A,B≤N)。

第二行为N个用空格隔开的非负整数，表示Ki​。

输出格式

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

输入输出样例

输入 #1复制

5 1 5
3 3 1 2 5

输出 #1复制

3

#include<cstdio>
#include<iostream>
using namespace std;
int n,a,b,cnt=0x7ffffff;//int的最大值
int to[205];
int vi[205];
void dfs(int step,int sum)//step表示当前搜到的楼层,sum表示按键次数
{
    if(step==b) cnt=min(cnt,sum);
    if(sum>cnt) return;
    vi[step]=1;//标记说明已经搜过 
    //如果没越界且没搜索过就搜索 
    if(step+to[step]<=n&&!vi[step+to[step]]) dfs(step+to[step],sum+1);
    if(step-to[step]>=1&&!vi[step-to[step]]) dfs(step-to[step],sum+1);
    vi[step]=0;//回溯
}
int main()
{
    scanf("%d%d%d",&n,&a,&b);
    for(int i=1;i<=n;i++) scanf("%d",&to[i]);
    vi[a]=1;
    dfs(a,0); 
    if(cnt!=0x7ffffff) printf("%d",cnt);
    else printf("-1");
    return 0;
}